khurrameycon commited on
Commit
5e5c791
·
verified ·
1 Parent(s): ea19a45

max limit set to 1024

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def predict_text(text):
66
  # inputs = processor(image, input_text, return_tensors="pt").to(device)
67
  inputs = processor(text=text, return_tensors="pt").to("cuda")
68
  # Generate a response from the model
69
- outputs = model.generate(**inputs, max_new_tokens=250)
70
 
71
  # Decode the output to return the final response
72
  response = processor.decode(outputs[0], skip_special_tokens=True)
 
66
  # inputs = processor(image, input_text, return_tensors="pt").to(device)
67
  inputs = processor(text=text, return_tensors="pt").to("cuda")
68
  # Generate a response from the model
69
+ outputs = model.generate(**inputs, max_new_tokens=1024)
70
 
71
  # Decode the output to return the final response
72
  response = processor.decode(outputs[0], skip_special_tokens=True)