MusIre commited on
Commit
9943577
·
verified ·
1 Parent(s): 15785c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -122,10 +122,10 @@ def generate_description(image):
122
  output = model_gptneo.generate(
123
  input_ids=input_ids,
124
  attention_mask=attention_mask,
125
- max_length=300,
126
  temperature=0.7,
127
  top_p=0.9,
128
- repetition_penalty=1.2,
129
  do_sample=True,
130
  pad_token_id=tokenizer.pad_token_id
131
  )
 
122
  output = model_gptneo.generate(
123
  input_ids=input_ids,
124
  attention_mask=attention_mask,
125
+ max_length=250,
126
  temperature=0.7,
127
  top_p=0.9,
128
+ repetition_penalty=1.5,
129
  do_sample=True,
130
  pad_token_id=tokenizer.pad_token_id
131
  )