Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 #gr.Interface.load("models/hipnologo/gpt2-imdb-finetune").launch()
+
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 def predict_review(text):
@@ -20,7 +21,19 @@ def predict_review(text):
     # getting the predicted class
     predicted_class = logits.argmax(-1).item()
 
-    return f"The sentiment predicted by the model is: {'Positive' if predicted_class == 1 else 'Negative'}"
+    sentiment = 'Positive' if predicted_class == 1 else 'Negative'
+
+    # Create a Markdown string for the output
+    result_md = f"## Sentiment: **{sentiment}**"
+    return result_md
 
-iface = gr.Interface(fn=predict_review, inputs="textbox", outputs="text")
+iface = gr.Interface(
+    fn=predict_review, 
+    inputs=gr.inputs.Textbox(lines=7, placeholder="Enter text here..."), 
+    outputs=gr.outputs.Markdown(),
+    title="Sentiment Analysis",
+    description="This application predicts the sentiment (Positive/Negative) of the input text using a fine-tuned GPT-2 model.",
+    theme="compact"  # change this to the theme you prefer: 'huggingface', 'default'
+)
 iface.launch()
+