Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
import gradio as gr
|
3 |
from transformers import pipeline
|
4 |
import re
|
5 |
-
|
6 |
import numpy as np
|
7 |
import pandas as pd
|
8 |
|
@@ -44,7 +44,7 @@ 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 =
|
48 |
|
49 |
scores = []
|
50 |
for rater in rater_models:
|
|
|
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 |
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:
|