Spaces:
Sleeping
Sleeping
hsuvaskakoty
commited on
Commit
•
4165ebd
1
Parent(s):
9a3ff71
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()}
|