Spaces:
Running
Running
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}"
|