Update app.py

app.py

@@ -27,6 +27,7 @@ def predict_text(params):
         return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
         
     texts = params.get("urls", [])
+    file_ids = params.get("normalfileID", [])
     # api = params.get("api", "")
     # job_id = params.get("job_id", "")
     print("JSON : \n",params)
@@ -35,7 +36,7 @@ def predict_text(params):
     
     solutions = []
     
-    for text in texts:
+    for text, file_id in zip(texts,file_ids):
         encoding = bert_tokenizer.encode_plus(
             text,
             add_special_tokens=True,
@@ -55,7 +56,7 @@ def predict_text(params):
         pred_label = tf.argmax(logits, axis=1).numpy()[0]
         
         label = {1: 'positive', 0: 'negative'}
-        result = {'text': text, 'label': [label[pred_label]]}
+        result = {'text': text, 'label': [label[pred_label]], "qcUser" : None, "file_id": file_id}
         solutions.append(result)
     
     # result_url = f"{api}/{job_id}"