Update app.py

app.py

@@ -1,16 +1,15 @@
 import gradio as gr
 #gr.Interface.load("models/hipnologo/gpt2-imdb-finetune").launch()
-import gradio as gr
 
-def postprocess_label(label):
-    if label == "LABEL_0":
+def postprocess_label(model, inp):
+    result = model(inp)
+    if result == "LABEL_0":
         return "Negative"
-    elif label == "LABEL_1":
+    elif result == "LABEL_1":
         return "Positive"
     else:
         return "Unknown label"
 
 iface = gr.Interface.load("models/hipnologo/gpt2-imdb-finetune")
-iface.update(outputs=gr.outputs.Textbox(label="Sentiment", postprocess=postprocess_label))
-
+iface.fn = lambda inp: postprocess_label(iface.fn, inp)
 iface.launch()