Update app.py
Browse files
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
|
11 |
-
return
|
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")
|