Update main.py

main.py

@@ -1,8 +1,8 @@
-from flask import Flask, jsonify, request
+
 from transformers import AutoAdapterModel, AutoTokenizer, TextClassificationPipeline
 from huggingface_hub import Repository
 
-app = Flask(__name__)
+# app = Flask(__name__)
 
 #define model
 tokenizer = AutoTokenizer.from_pretrained("UBC-NLP/MARBERT")
@@ -22,15 +22,15 @@ model.load_adapter_fusion("/fusion_adapter/aoc(3),aoc(4),sarcasm",with_head=True
 
 pipe = TextClassificationPipeline(tokenizer=tokenizer, model=model)
 
-@app.route('/predict', methods=['POST'])
-def predict():
-  text = request.json['inputs']
+# @app.route('/predict', methods=['POST'])
+# def predict():
+#   text = request.json['inputs']
 
-  prediction = pipe(text)
-  labels = {"LABEL_0":"GULF", "LABEL_1":"LEVANT","LABEL_2":"EGYPT"}
-  regions = []
-  for res in prediction:
-    regions.append(labels[res['label']])
+#   prediction = pipe(text)
+#   labels = {"LABEL_0":"GULF", "LABEL_1":"LEVANT","LABEL_2":"EGYPT"}
+#   regions = []
+#   for res in prediction:
+#     regions.append(labels[res['label']])
 
-  return jsonify({'response': regions})
+#   return jsonify({'response': regions})