Update app.py
Browse files
app.py
CHANGED
@@ -20,10 +20,12 @@ def model_to_cuda(model):
|
|
20 |
else:
|
21 |
print("CUDA not available! Using CPU.")
|
22 |
return model
|
|
|
23 |
@spaces.GPU
|
24 |
def translate_text(input_text, sselected_language, tselected_language, model_name):
|
25 |
sl = all_langs[sselected_language]
|
26 |
tl = all_langs[tselected_language]
|
|
|
27 |
if model_name == "Helsinki-NLP":
|
28 |
try:
|
29 |
model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
|
|
|
20 |
else:
|
21 |
print("CUDA not available! Using CPU.")
|
22 |
return model
|
23 |
+
|
24 |
@spaces.GPU
|
25 |
def translate_text(input_text, sselected_language, tselected_language, model_name):
|
26 |
sl = all_langs[sselected_language]
|
27 |
tl = all_langs[tselected_language]
|
28 |
+
message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
|
29 |
if model_name == "Helsinki-NLP":
|
30 |
try:
|
31 |
model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
|