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")
|