space updated

app.py

@@ -14,7 +14,7 @@ def translate(text, device, max_length, num_beams, translation_way = "en-lt"):
     }
     if translation_way not in translations_ways:
         raise ValueError(f"Invalid translation way. Supported ways: {list(translations_ways.keys())}")
-    text = f"{translations_ways[translation_way]} {text}"
+    input_text = f"{translations_ways[translation_way]} {text}"
     encoded_input = TOKENIZER(input_text, return_tensors="pt", padding=True, truncation=True, max_length=128).to(device)
     with torch.no_grad():
         output_tokens = MODEL.generate(