Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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=
|
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,
|