Spaces:
Runtime error
Runtime error
c
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -20,8 +20,8 @@ tokenizer = AutoTokenizer.from_pretrained(
|
|
20 |
)
|
21 |
|
22 |
ft_model.eval()
|
23 |
-
def greet(
|
24 |
-
model_input = tokenizer(
|
25 |
with torch.no_grad():
|
26 |
generation = ft_model.generate(**model_input, max_new_tokens = 150)
|
27 |
result = tokenizer.decode(generation[0], skip_special_tokens=True)
|
|
|
20 |
)
|
21 |
|
22 |
ft_model.eval()
|
23 |
+
def greet(your_prompt):
|
24 |
+
model_input = tokenizer(your_prompt, return_tensors="pt").to("cpu")
|
25 |
with torch.no_grad():
|
26 |
generation = ft_model.generate(**model_input, max_new_tokens = 150)
|
27 |
result = tokenizer.decode(generation[0], skip_special_tokens=True)
|