Update app.py

app.py

@@ -31,12 +31,12 @@ user_friendly_name_list = list(user_friendly_name.values())
 #demo = gr.Interface.from_pipeline(pipe)
 #demo.launch()
 
-def predict(text, chosen_model_name):    
+def predict(text, chosen_model):    
 
-    chosen_model = reverse_user_friendly_name[chosen_model_name]
+    chosen_model_name = reverse_user_friendly_name[chosen_model]
     
     # Initialize the pipeline with the chosen model
-    model_pipeline = pipeline("text-classification", model=chosen_model)
+    model_pipeline = pipeline("text-classification", model=chosen_model_name)
     result = model_pipeline(text)
     label = result[0]['label']