Update app.py

app.py

@@ -83,12 +83,25 @@ aspect_aliases = {
     "value": "cena", "experience": "doświadczenie", "recommendation": "ogólna ocena",
     "children": "dzieci", "family": "rodzina", "pet": "zwierzęta"
 }
-def translate_pl_to_en(texts):
-    return [res["translation_text"] for res in pl_to_en(texts)]
+# ───────────────────── tłumaczenia  ──────────────────────
+def translate_pl_to_en(texts: list[str]) -> list[str]:
+    inputs = pl_to_en_tok(texts,
+                          return_tensors="pt",
+                          padding=True,
+                          truncation=True).to(device)
+    with torch.no_grad():
+        generated = pl_to_en_mod.generate(**inputs)
+    return pl_to_en_tok.batch_decode(generated, skip_special_tokens=True)
 
 
-def translate_en_to_pl(texts):
-    return [res["translation_text"] for res in en_to_pl(texts)]
+def translate_en_to_pl(texts: list[str]) -> list[str]:
+    inputs = en_to_pl_tok(texts,
+                          return_tensors="pt",
+                          padding=True,
+                          truncation=True).to(device)
+    with torch.no_grad():
+        generated = en_to_pl_mod.generate(**inputs)
+    return en_to_pl_tok.batch_decode(generated, skip_special_tokens=True)
 
 
 def extract_aspects(text_en: str):