Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,6 @@
|
|
| 2 |
import gradio as gr
|
| 3 |
from transformers import pipeline
|
| 4 |
import re
|
| 5 |
-
import langid
|
| 6 |
import numpy as np
|
| 7 |
import pandas as pd
|
| 8 |
|
|
@@ -44,7 +43,6 @@ def run_benchmark():
|
|
| 44 |
gen_output = gen_model(prompt, max_new_tokens=100, do_sample=True)[0]['generated_text'].strip()
|
| 45 |
valid = is_palindrome(gen_output)
|
| 46 |
cleaned_len = len(clean_text(gen_output))
|
| 47 |
-
detected_lang = langid.classify(gen_output)[0]
|
| 48 |
|
| 49 |
scores = []
|
| 50 |
for rater in rater_models:
|
|
@@ -63,8 +61,7 @@ def run_benchmark():
|
|
| 63 |
"Valid": "✅" if valid else "❌",
|
| 64 |
"Length": cleaned_len,
|
| 65 |
"Grammar Score": avg_score,
|
| 66 |
-
"Final Score": final_score
|
| 67 |
-
"Detected Lang": detected_lang
|
| 68 |
})
|
| 69 |
|
| 70 |
df = pd.DataFrame(results).sort_values(by="Final Score", ascending=False).reset_index(drop=True)
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
from transformers import pipeline
|
| 4 |
import re
|
|
|
|
| 5 |
import numpy as np
|
| 6 |
import pandas as pd
|
| 7 |
|
|
|
|
| 43 |
gen_output = gen_model(prompt, max_new_tokens=100, do_sample=True)[0]['generated_text'].strip()
|
| 44 |
valid = is_palindrome(gen_output)
|
| 45 |
cleaned_len = len(clean_text(gen_output))
|
|
|
|
| 46 |
|
| 47 |
scores = []
|
| 48 |
for rater in rater_models:
|
|
|
|
| 61 |
"Valid": "✅" if valid else "❌",
|
| 62 |
"Length": cleaned_len,
|
| 63 |
"Grammar Score": avg_score,
|
| 64 |
+
"Final Score": final_score
|
|
|
|
| 65 |
})
|
| 66 |
|
| 67 |
df = pd.DataFrame(results).sort_values(by="Final Score", ascending=False).reset_index(drop=True)
|