Update app.py

app.py

@@ -66,22 +66,28 @@ with app:
     Model Predictions
     """)
 
-    # with gr.Row():
-    #   model1_input = gr.Textbox(label="Model 1")
+    gr.Markdown(
+    """
+    Model 1 = juliensimon/distilbert-amazon-shoe-reviews
+    """)
+
     with gr.Row():
-        btn1 = gr.Button("Prediction for Model 1")
+        btn1 = gr.Button("Predict for Model 1")
     with gr.Row():
-        out_1 = gr.Textbox(label="Prediction")
+        out_1 = gr.Textbox(label="Prediction for Model 1")
 
     classifier = pipeline("text-classification", model=model_id_1)   
     btn1.click(fn=predict, inputs=inp_1, outputs=out_1)
 
-    # with gr.Row():
-    #   model2_input = gr.Textbox(label="Model 2") 
+    gr.Markdown(
+    """
+    Model 2 = juliensimon/distilbert-amazon-shoe-reviews
+    """)
+
     with gr.Row():
-        btn2 = gr.Button("Prediction for Model 2")
+        btn2 = gr.Button("Predict for Model 2")
     with gr.Row():
-        out_2 = gr.Textbox(label="Prediction")   
+        out_2 = gr.Textbox(label="Prediction for Model 2")   
     classifier = pipeline("text-classification", model=model_id_2)    
     btn2.click(fn=predict, inputs=inp_1, outputs=out_2)