Spaces:
Build error
Build error
Ravi theja K
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 24 |
|
| 25 |
content = "5x + 35 = 7x - 60 + 10. Solve for x"
|
| 26 |
messages = [{"role": "user", "content": content}]
|
| 27 |
-
input_ids =
|
| 28 |
|
| 29 |
outputs = model.generate(input_ids=input_ids, max_new_tokens=256)
|
| 30 |
print(Llamatokenizer.decode(outputs[0]))
|
|
|
|
| 24 |
|
| 25 |
content = "5x + 35 = 7x - 60 + 10. Solve for x"
|
| 26 |
messages = [{"role": "user", "content": content}]
|
| 27 |
+
input_ids = Llamatokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to("cuda")
|
| 28 |
|
| 29 |
outputs = model.generate(input_ids=input_ids, max_new_tokens=256)
|
| 30 |
print(Llamatokenizer.decode(outputs[0]))
|