bdsqlsz commited on
Commit
a32cba6
·
verified ·
1 Parent(s): ab1f7d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def run_example(image):
62
  )
63
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
64
  parsed_answer = processor.post_process_generation(generated_text, task=task_prompt, image_size=(image.width, image.height))
65
- return modify_caption(parsed_answer["<DESCRIPTION>"])
66
 
67
 
68
  css = """
 
62
  )
63
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
64
  parsed_answer = processor.post_process_generation(generated_text, task=task_prompt, image_size=(image.width, image.height))
65
+ return modify_caption(parsed_answer["<MORE_DETAILED_CAPTION>"])
66
 
67
 
68
  css = """