Update app.py

app.py

@@ -26,6 +26,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_id = params.get("normalfileID",[])
     # api = params.get("api", "")
     # job_id = params.get("job_id","")
 
@@ -34,7 +35,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,
@@ -54,7 +55,7 @@ def predict_text(params):
         pred_label = tf.argmax(logits, axis=1).numpy()[0]
         
         label = {0: 'BUSINESS', 1: 'COMEDY', 2: 'CRIME', 3: 'FOOD & DRINK', 4: 'POLITICS', 5: 'SPORTS', 6: 'TRAVEL'}
-        result = {'text':text, 'label':[label[pred_label]]}
+        result = {'text':text, 'label':[label[pred_label]], "qcUser" : None,"normalfileID":file_id}
         solutions.append(result)
 
     # result_url = f"{api}/{job_id}"