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