Spaces:
Running
Running
Upload folder using huggingface_hub
Browse files- main/model_inference.py +1 -1
main/model_inference.py
CHANGED
|
@@ -26,7 +26,7 @@ def api(payload: InputData):
|
|
| 26 |
|
| 27 |
tweet = payload.comment
|
| 28 |
model_input = format_model_input(tweet)
|
| 29 |
-
model_response = model.predict(model_input)
|
| 30 |
|
| 31 |
label = int(model_response["class_label"][0])
|
| 32 |
probability_scores = model_response["class_probability_scores"]
|
|
|
|
| 26 |
|
| 27 |
tweet = payload.comment
|
| 28 |
model_input = format_model_input(tweet)
|
| 29 |
+
model_response = model.predict(context=None, model_input=model_input)
|
| 30 |
|
| 31 |
label = int(model_response["class_label"][0])
|
| 32 |
probability_scores = model_response["class_probability_scores"]
|