sohomghosh commited on
Commit
a186519
1 Parent(s): f32f265

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ model_read.load_state_dict(torch.load('readability_model.bin', map_location=devi
19
 
20
 
21
  def get_readability(text):
22
- df = pd.DataFrame({'sentence':text})
23
  actual_predictions_read = do_predict(model_read, tokenizer_read, df)
24
  score = round(actual_predictions_read[1][0], 4)
25
  return score
 
19
 
20
 
21
  def get_readability(text):
22
+ df = pd.DataFrame({'sentence':[text]})
23
  actual_predictions_read = do_predict(model_read, tokenizer_read, df)
24
  score = round(actual_predictions_read[1][0], 4)
25
  return score