Fix app

app.py

@@ -4,12 +4,16 @@ from transformers import pipeline
 class UI:
     def __init__(self):
         self.model = pipeline("sentiment-analysis", model="lyrisha/bert-finetuned-imdb-sentiment")
+        
+    def predict(text):
+    	predictions = self.model(text)
+    	return {p["label"]: p["score"] for p in predictions}
 
     def launch(self):
         demo = gr.Interface(
-            fn=self.model,
+            fn=self.predict,
             inputs="textbox",
-            outputs="text",
+            outputs=gr.Label(num_top_classes=2),
             title="Check the mood of the sentence",
             description="The model was trained on 'imdb' dataset to classify sentences as 'positive' or 'negative' attitude.",
             examples=[["Great news! My reality check just bounced."],