Update app.py

app.py

@@ -13,7 +13,6 @@ if hf_token is None:
 # 🔹 Charger le tokenizer et le modèle
 model_id = "SafaaAI/final_llm_darija_fr_tech"
 
-# Utiliser AutoTokenizer directement
 tokenizer = AutoTokenizer.from_pretrained(
     model_id,
     token=hf_token,
@@ -41,11 +40,14 @@ def chat_with_model(message, history):
     
     # Encoder le prompt avec le tokenizer
     inputs = tokenizer(full_prompt, return_tensors="pt").to(model.device)
-
+    
+    # 📝 Passer les arguments explicitement
+    input_ids = inputs["input_ids"]
+    
     # Générer la réponse
     with torch.no_grad():
         output_ids = model.generate(
-            **inputs,
+            input_ids,
             max_new_tokens=200,
             do_sample=True,
             top_p=0.9,