MusIre commited on
Commit
5a0ed9f
·
verified ·
1 Parent(s): 1bcbacf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def generate_description(image_path):
139
  input_ids = tokenizer.encode(full_prompt, return_tensors="pt").to(device)
140
  output = model_gptneo.generate(
141
  input_ids=input_ids,
142
- max_length=350,
143
  num_return_sequences=1,
144
  temperature=0.7,
145
  top_p=0.9,
 
139
  input_ids = tokenizer.encode(full_prompt, return_tensors="pt").to(device)
140
  output = model_gptneo.generate(
141
  input_ids=input_ids,
142
+ max_length=250,
143
  num_return_sequences=1,
144
  temperature=0.7,
145
  top_p=0.9,