Update app.py

app.py

@@ -1,6 +1,4 @@
 
-# https://huggingface.co/spaces/CK42/sentiment-model-comparison/blob/main/app.py
-
 # import sklearn
 from os import O_ACCMODE
 import gradio as gr
@@ -43,8 +41,6 @@ def get_metadata(model_id):
         return metadata
     except requests.exceptions.HTTPError:
         return None
-
-# classifier = pipeline("text-classification", model="juliensimon/distilbert-amazon-shoe-reviews")
                 
 def predict(review, model_id):
         classifier = pipeline("text-classification", model=model_id)
@@ -66,11 +62,10 @@ with app:
         inp_1= gr.Textbox(label="Type text here.",placeholder="The customer service was satisfactory.")
         out_2 = gr.Textbox(label="Prediction")
 
-    
-    # gr.Markdown(
-    # """
-    # Model Predictions
-    # """)
+     gr.Markdown(
+     """
+     Model Predictions
+     """)
     with gr.Row():
       model1_input = gr.Textbox(label="Model 1")
     with gr.Row():
@@ -83,7 +78,7 @@ with app:
       model2_input = gr.Textbox(label="Model 2") 
     with gr.Row():
         btn = gr.Button("Prediction for Model 2")
-    btn.click(fn=predict(model_id_2), inputs=inp_1, outputs=out_2)
+    btn.click(fn=load_agent(model_id_2), inputs=inp_1, outputs=out_2)
 
 
     # app_button.click(load_agent, inputs=[model1_input, model2_input], outputs=[model1_name, model1_score_output, model2_name, model2_score_output])