-

app.py

@@ -40,7 +40,9 @@ def chatbot_response(user_input):
         return "أنا متخصص فقط في النخيل والتمور. يرجى طرح أسئلة ذات صلة."
 
     prompt = f"سؤال: {user_input}\nجواب:"
-    inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+    inputs = tokenizer(prompt, return_tensors="pt")
+    inputs = {k: v.to(model.device) for k, v in inputs.items()}
+
 
     outputs = model.generate(
         **inputs,