llepogam commited on
Commit
e9c60fb
·
1 Parent(s): 9b6bc2c

hopefully last commit

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 = [preprocessed_result['text_clean']['0']]
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'