Update app.py

app.py

@@ -46,13 +46,11 @@ def get_prediction(model_id):
              
     def predict(review):
             prediction = classifier(review)
-            predictions = []
-            for p in prediction:
-                new_pred = print(p)
-                predictions.append(new_pred)
-            return predictions
+            print(prediction)
+            return prediction
     return predict
 
+
 with app:
     gr.Markdown(
     """
@@ -101,8 +99,5 @@ with app:
         out_3 = gr.Textbox(label="Prediction for Model 3")   
     classifier = pipeline("text-classification", model=model_id_3)    
     btn3.click(fn=get_prediction(model_id_2), inputs=inp_1, outputs=out_3)
-
-
-
-    
+   
 app.launch()