Update app.py

Added label description

app.py

@@ -1,15 +1,23 @@
 import gradio as gr
 #gr.Interface.load("models/hipnologo/gpt2-imdb-finetune").launch()
 
-def postprocess_label(model, inp):
-    result = model(inp)
-    if result == "LABEL_0":
-        return "Negative"
-    elif result == "LABEL_1":
-        return "Positive"
-    else:
-        return "Unknown label"
-
-iface = gr.Interface.load("models/hipnologo/gpt2-imdb-finetune")
-iface.fn = lambda inp: postprocess_label(iface.fn, inp)
+import gradio as gr
+from transformers import pipeline
+
+# Load the sentiment analysis pipeline
+sentiment_analysis = pipeline("text-classification", model="models/hipnologo/gpt2-imdb-finetune")
+
+def predict(review_text):
+    # Run the review text through the pipeline
+    result = sentiment_analysis(review_text)[0]
+    
+    # Map the LABEL_0/LABEL_1 to Negative/Positive
+    if result['label'] == 'LABEL_0':
+        result['label'] = 'Negative'
+    elif result['label'] == 'LABEL_1':
+        result['label'] = 'Positive'
+    
+    return result['label'], float(result['score'])
+
+iface = gr.Interface(fn=predict, inputs="textbox", outputs=["label", "number"])
 iface.launch()