Update app.py

app.py

@@ -41,7 +41,7 @@ user_friendly_name_list = list(user_friendly_name.values())
 def predict(text, chosen_model):    
     
     # Initialize the pipeline with the chosen model
-    model_pipeline = pipeline("text-classification", model=reverse_user_friendly_name[chosen_model])
+    model_pipeline = pipeline("text-classification", model=chosen_model)
     result = model_pipeline(text)
     label = result[0]['label']