Update app.py

app.py

@@ -54,6 +54,15 @@ def get_metadata(model_id):
     except requests.exceptions.HTTPError:
         return None
 
+classifier = pipeline("text-classification", model="juliensimon/distilbert-amazon-shoe-reviews")
+                
+def predict(review):
+        prediction = classifier(review)
+        print(prediction)
+        stars = prediction[0]['label']
+        stars = (int)(stars.split('_')[1])+1
+        score = 100*prediction[0]['score']
+        return "{} {:.0f}%".format("\U00002B50"*stars, score)
 
 with app:
     gr.Markdown(
@@ -70,7 +79,7 @@ with app:
         inp = gr.Textbox(placeholder="The customer service was satisfactory.")
         out = gr.Textbox()
     btn = gr.Button("Run")
-    btn.click(fn=update, inputs=inp, outputs=out)
+    btn.click(fn=predict, inputs=inp, outputs=out)
      
     with gr.Row():
       app_button = gr.Button("Compare models")