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,
|