Update app.py

app.py

@@ -27,7 +27,7 @@ MODEL_NAME = "Qwen/Qwen3-0.6B"
 
 
 def load_model():
-    model = AutoModel.from_pretrained(MODEL_NAME).to(device)
+    model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to(device)
     print(f"Model loaded in {device}")
     return model
 
@@ -63,7 +63,7 @@ def _translate(text: str, src_lang: str, tgt_lang: str):
             .numpy()
             .tolist()
         )
-        translated_chunk = model(
+        translated_chunk = model.generate(
             input_ids=torch.tensor([input_tokens]).to(device),
             max_length=len(input_tokens) + 10000,
             num_return_sequences=1,