Update app.py

app.py

@@ -16,6 +16,7 @@ def Multi_Label_Classification_of_Pubmed_Articles(model_input: str) -> Dict[str,
     
     # tokenized = tokenizer.tokenize_and_pad(model_input)
     # Encoding input data
+    max_length = 128
     Articles_test = list(model_input)
     test_encodings = tokenizer.batch_encode_plus(Articles_test,max_length=max_length,padding=True,truncation=True)
     test_input_ids = test_encodings['input_ids']
@@ -63,12 +64,12 @@ def Multi_Label_Classification_of_Pubmed_Articles(model_input: str) -> Dict[str,
     true_bools = [tl==1 for tl in true_labels]
     
 
-    prediction = model.predict(tokenized)[0]
-    ret = {
-        "negative": float(prediction[0]),
-        "positive": float(prediction[1])
-    }
-    return ret
+    #prediction = model.predict(tokenized)[0]
+    #ret = {
+    #    "negative": float(prediction[0]),
+    #    "positive": float(prediction[1])
+    #}
+    return true_bools
 
 
 model_input = gr.Textbox("Input text here", show_label=False)