Spaces:
Sleeping
Sleeping
RaghadAbdulaziz
commited on
Commit
·
7893b49
1
Parent(s):
021194e
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,
|