c commited on
Commit
c1cd225
·
verified ·
1 Parent(s): 04a2b0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ tokenizer = AutoTokenizer.from_pretrained(
20
  )
21
 
22
  ft_model.eval()
23
- def greet(eval_prompt):
24
- model_input = tokenizer(eval_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)
 
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)