yiyii commited on
Commit
7af90dd
1 Parent(s): 6eefb37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,7 +98,7 @@ client = InferenceClient(
98
  # "mistralai/Mistral-7B-v0.1"
99
  # )
100
 
101
- def generate(image, temperature=0.9, max_new_tokens=3000, top_p=0.95, repetition_penalty=1.0):
102
  image_caption, gender, age, emotion = get_image_info(image)
103
  print("............................................")
104
  print("image_caption:", image_caption)
@@ -195,7 +195,7 @@ demo = gr.Interface(fn=generate,
195
 
196
  gr.Slider(
197
  label="Max new tokens",
198
- value=3000,
199
  minimum=0,
200
  maximum=3000,
201
  step=1.0,
 
98
  # "mistralai/Mistral-7B-v0.1"
99
  # )
100
 
101
+ def generate(image, temperature=0.9, max_new_tokens=1500, top_p=0.95, repetition_penalty=1.0):
102
  image_caption, gender, age, emotion = get_image_info(image)
103
  print("............................................")
104
  print("image_caption:", image_caption)
 
195
 
196
  gr.Slider(
197
  label="Max new tokens",
198
+ value=1500,
199
  minimum=0,
200
  maximum=3000,
201
  step=1.0,