Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -47,12 +47,15 @@ def get_prediction(model_id):
|
|
47 |
|
48 |
def predict(review):
|
49 |
prediction = classifier(review)
|
50 |
-
|
|
|
|
|
|
|
51 |
# stars = prediction[0]['label']
|
52 |
# stars = (int)(stars.split('_')[1])+1
|
53 |
# score = 100*prediction[0]['score']
|
54 |
# return "{} {:.0f}%".format("\U00002B50"*stars, score)
|
55 |
-
return
|
56 |
return predict
|
57 |
|
58 |
with app:
|
|
|
47 |
|
48 |
def predict(review):
|
49 |
prediction = classifier(review)
|
50 |
+
predictions = []
|
51 |
+
for p in prediction:
|
52 |
+
new_pred = print(prediction)
|
53 |
+
predictions.append(new_pred)
|
54 |
# stars = prediction[0]['label']
|
55 |
# stars = (int)(stars.split('_')[1])+1
|
56 |
# score = 100*prediction[0]['score']
|
57 |
# return "{} {:.0f}%".format("\U00002B50"*stars, score)
|
58 |
+
return predictions
|
59 |
return predict
|
60 |
|
61 |
with app:
|