Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,6 @@ def sentiment_analysis(payload: ClassificationRequest):
|
|
| 29 |
predictions = model.predict(X)
|
| 30 |
pred_prob = model.predict_proba(X)
|
| 31 |
|
| 32 |
-
|
| 33 |
return {
|
| 34 |
category_list[predictions.tolist()[0]]: pred_prob[predictions.tolist()[0]],
|
| 35 |
category_list[0]: pred_prob[0],
|
|
@@ -37,5 +36,5 @@ def sentiment_analysis(payload: ClassificationRequest):
|
|
| 37 |
category_list[2]: pred_prob[2],
|
| 38 |
category_list[3]: pred_prob[3],
|
| 39 |
category_list[4]: pred_prob[4]
|
| 40 |
-
}
|
| 41 |
|
|
|
|
| 29 |
predictions = model.predict(X)
|
| 30 |
pred_prob = model.predict_proba(X)
|
| 31 |
|
|
|
|
| 32 |
return {
|
| 33 |
category_list[predictions.tolist()[0]]: pred_prob[predictions.tolist()[0]],
|
| 34 |
category_list[0]: pred_prob[0],
|
|
|
|
| 36 |
category_list[2]: pred_prob[2],
|
| 37 |
category_list[3]: pred_prob[3],
|
| 38 |
category_list[4]: pred_prob[4]
|
| 39 |
+
}
|
| 40 |
|