hsuvaskakoty commited on
Commit
6c89cc9
·
verified ·
1 Parent(s): 253d03c

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,7 +17,8 @@ model_dict = {
17
  "Offensive Language Detection": "offensive"
18
  }
19
 
20
- def process_url(url, model_name):
 
21
  processed_text = data_prep.process_data(url)
22
  final_scores = model_predict.predict_text(processed_text, model_name)
23
 
 
17
  "Offensive Language Detection": "offensive"
18
  }
19
 
20
+ def process_url(url, model_key):
21
+ model_name = model_dict[model_key]
22
  processed_text = data_prep.process_data(url)
23
  final_scores = model_predict.predict_text(processed_text, model_name)
24