Update app.py

app.py

@@ -19,11 +19,12 @@ model_list = [
 #demo = gr.Interface.from_pipeline(pipe)
 #demo.launch()
 
-def predict(chosen_model):
+def predict(text, chosen_model):
 
     # Initialize the pipeline with the chosen model
     model_pipeline = pipeline("text-classification", model=chosen_model)
-    label = model_pipeline[0]['label']
+    result = model_pipeline(text)
+    label = result[0]['label']
     return label
 
 inputs = [