Tsunnami commited on
Commit
0436b7b
·
verified ·
1 Parent(s): 292e32a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ model = load_from_checkpoint(model_path)
7
  def score_translation(src_text, mt_text):
8
  translations = [{"src": src_text, "mt": mt_text}]
9
  results = model.predict(translations, batch_size=1, gpus=1)
10
- scores = results.get('scores', [[]])[0]
11
- return {f"Score {i+1}": score for i, score in enumerate(scores)}
12
 
13
  with gr.Blocks() as demo:
14
  gr.Markdown("# Translation Quality Scoring with ComeTH Model")
 
7
  def score_translation(src_text, mt_text):
8
  translations = [{"src": src_text, "mt": mt_text}]
9
  results = model.predict(translations, batch_size=1, gpus=1)
10
+ scores = results['scores'][0]
11
+ return scores
12
 
13
  with gr.Blocks() as demo:
14
  gr.Markdown("# Translation Quality Scoring with ComeTH Model")