Spaces:
Build error
Build error
hopefully last commit
Browse files
app.py
CHANGED
|
@@ -113,8 +113,9 @@ async def predict(predictionFeatures: PredictionFeatures):
|
|
| 113 |
# Perform prediction
|
| 114 |
preprocessed_result = loaded_model.predict(pd.DataFrame(df))
|
| 115 |
|
|
|
|
| 116 |
|
| 117 |
-
list_text = [
|
| 118 |
|
| 119 |
# Load model from MLflow
|
| 120 |
logged_model = 'runs:/227d2f8e431d40d6b5231add3a00d048/hate_speech_detection'
|
|
|
|
| 113 |
# Perform prediction
|
| 114 |
preprocessed_result = loaded_model.predict(pd.DataFrame(df))
|
| 115 |
|
| 116 |
+
df_preprocessed = pd.DataFrame(preprocessed_result)
|
| 117 |
|
| 118 |
+
list_text = [df_preprocessed.loc[0,"text_clean"]]
|
| 119 |
|
| 120 |
# Load model from MLflow
|
| 121 |
logged_model = 'runs:/227d2f8e431d40d6b5231add3a00d048/hate_speech_detection'
|