JoannaKOKO commited on
Commit
3b724fd
·
verified ·
1 Parent(s): 0dcb9ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ Provide a professional interpretation covering:
93
  - Practical advice
94
  - Potential outcomes"""
95
  input_ids = tokenizer(prompt, return_tensors="pt").to("cuda")
96
- outputs = model.generate(**input_ids, max_new_tokens=32)
97
  # Move output to CPU before decoding
98
  interpretation = tokenizer.decode(outputs[0].cpu(), skip_special_tokens=True)
99
  return interpretation
 
93
  - Practical advice
94
  - Potential outcomes"""
95
  input_ids = tokenizer(prompt, return_tensors="pt").to("cuda")
96
+ outputs = model.generate(**input_ids, max_new_tokens=500)
97
  # Move output to CPU before decoding
98
  interpretation = tokenizer.decode(outputs[0].cpu(), skip_special_tokens=True)
99
  return interpretation