Update app.py

app.py

@@ -10,7 +10,7 @@ def greet(name):
 model = SetFitModel.from_pretrained("dvilasuero/setfit-mini-imdb")
 
 def classify(text):
-    return model(text)
+    return model([text])[0]
     
 iface = gr.Interface(fn=classify, inputs="text", outputs="text")
 iface.launch()