Update app.py

app.py

@@ -44,9 +44,10 @@ def sentiment_analysis(text):
     X_input_ids, X_attention_mask = preprocess_input_data(text, arabert_tokenizer)
     predictions = model(X_input_ids)
     a=predictions.numpy()
-    return a
+    import numpy as np
+    predicted_class=list(np.where(preds <0.5,0,1).reshape(len(preds),1))
 
-import gradio as gr
+    return predicted_class
 
 
 iface = gr.Interface(fn=sentiment_analysis, inputs="text", outputs="text")