PabloTJ commited on
Commit
ed4ee58
·
verified ·
1 Parent(s): 15da496

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@
2
  import gradio as gr
3
  from transformers import pipeline
4
  import re
5
- from langdetect import detect
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 = detect(gen_output)
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: