Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ model_dict = {
|
|
| 12 |
def process_url(url, model_key):
|
| 13 |
model_name = model_dict[model_key]
|
| 14 |
processed_text = data_prep.process_data(url)
|
| 15 |
-
final_scores
|
| 16 |
highest_prob_label = max(final_scores, key=final_scores.get)
|
| 17 |
highest_prob = final_scores[highest_prob_label]
|
| 18 |
progress_bars = {label: score for label, score in final_scores.items()}
|
|
|
|
| 12 |
def process_url(url, model_key):
|
| 13 |
model_name = model_dict[model_key]
|
| 14 |
processed_text = data_prep.process_data(url)
|
| 15 |
+
final_scores = model_predict.predict_text(processed_text, model_name)
|
| 16 |
highest_prob_label = max(final_scores, key=final_scores.get)
|
| 17 |
highest_prob = final_scores[highest_prob_label]
|
| 18 |
progress_bars = {label: score for label, score in final_scores.items()}
|