Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,8 @@ model_dict = {
|
|
17 |
"Offensive Language Detection": "offensive"
|
18 |
}
|
19 |
|
20 |
-
def process_url(url,
|
|
|
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 |
|