Update app.py

app.py

@@ -23,8 +23,11 @@ def predict_label(text):
     
     ent_scores = extract_ent_scores(entity_model,ip,ensemble_pooled_scores)
     combined_sequences, ent_pred_tags = pool_ent_scores(ent_scores, ip_len)
+
+    
+    ops = [[i,o] for i,o in zip(ip,combined_sequences[-1]]
     
-    return combined_sequences
+    return ops
 
 
 if __name__ == '__main__':
@@ -66,13 +69,14 @@ if __name__ == '__main__':
         example_input=gr.Textbox(label="Input Example", lines=3)
         prediction=gr.Text(label="Predicted Entities")
         
+        
         gr.Interface(fn=predict_label, inputs=example_input,
                     outputs=prediction,theme="smooth_slate",
-                    title="Flat Entity Classification for Levant Arabic")
+                    title="Flat Entity Classification for Levantine Arabic")
         gr.Examples(
       examples=["النشرة الإخبارية الصادرة عن الأونروا رقم 113 (1986/1/8).",
                "صورة لمدينة أريحا القديمة :تل السلطان",
-               "صورة اطفال مخيم للاجئين الفلسطينيين ي لبنان"],
+               "صورة اطفال مخيم للاجئين الفلسطينيين في لبنان"],
       inputs= example_input)
         
     iface.launch(show_api=False)