RaghadAbdulaziz commited on
Commit
7893b49
·
1 Parent(s): 021194e
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -40,9 +40,7 @@ def chatbot_response(user_input):
40
  return "أنا متخصص فقط في النخيل والتمور. يرجى طرح أسئلة ذات صلة."
41
 
42
  prompt = f"سؤال: {user_input}\nجواب:"
43
- inputs = tokenizer(prompt, return_tensors="pt")
44
- inputs = {k: v.to(model.device) for k, v in inputs.items()}
45
-
46
 
47
  outputs = model.generate(
48
  **inputs,
 
40
  return "أنا متخصص فقط في النخيل والتمور. يرجى طرح أسئلة ذات صلة."
41
 
42
  prompt = f"سؤال: {user_input}\nجواب:"
43
+ inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
 
 
44
 
45
  outputs = model.generate(
46
  **inputs,