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 |
|