Update app.py

app.py

@@ -1,3 +1,20 @@
 import gradio as gr
 
-gr.load("models/quocviethere/roberta-sentiment-model").launch()
+def predict_sentiment(text):
+    model_output = gr.load("models/quocviethere/roberta-sentiment-model").process(text)
+    # Convert the model output to a more readable format
+    if model_output[0] > model_output[1]:  # Assuming index 0 is for negative sentiment
+        return "Negative Review", model_output[0]
+    else:
+        return "Positive Review", model_output[1]
+
+iface = gr.Interface(
+    fn=predict_sentiment,
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Type your review here"),
+    outputs=[
+        gr.outputs.Label(num_top_classes=2, label="Sentiment Analysis")
+    ]
+)
+
+iface.launch()
+