Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -34,10 +34,10 @@ def generate_caption(image):
|
|
34 |
generated_text = florence_processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
|
35 |
parsed_answer = florence_processor.post_process_generation(
|
36 |
generated_text,
|
37 |
-
task="<
|
38 |
image_size=(image.width, image.height)
|
39 |
)
|
40 |
-
prompt = parsed_answer["<
|
41 |
print("\nGeneration completed!:"+ prompt)
|
42 |
return prompt
|
43 |
# yield prompt, None
|
|
|
34 |
generated_text = florence_processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
|
35 |
parsed_answer = florence_processor.post_process_generation(
|
36 |
generated_text,
|
37 |
+
task="<DETAILED_CAPTION>",
|
38 |
image_size=(image.width, image.height)
|
39 |
)
|
40 |
+
prompt = parsed_answer["<DETAILED_CAPTION>"]
|
41 |
print("\nGeneration completed!:"+ prompt)
|
42 |
return prompt
|
43 |
# yield prompt, None
|