Update app.py

app.py

@@ -7,8 +7,12 @@ tokenizer = AutoTokenizer.from_pretrained('SeyedAli/Persian-Text-Sentiment-Bert-
 
 def Sentiment(text):
     pipline = pipeline(task="text-classification", model=model, tokenizer=tokenizer)
-    output=pipline(text)
-    return output
+    preds=pipline(text)
+    # return output
+    outputs = {}
+    for p in preds:
+        outputs[p["label"]] = p["score"]
+    return outputs
 
-iface = gr.Interface(fn=Sentiment, inputs="text", outputs="text")
+iface = gr.Interface(fn=Sentiment, inputs="text", outputs=gr.outputs.Label())
 iface.launch(share=False)