EmoCube commited on
Commit
5f6fb88
·
verified ·
1 Parent(s): 3e84585

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="<MORE_DETAILED_CAPTION>",
38
  image_size=(image.width, image.height)
39
  )
40
- prompt = parsed_answer["<MORE_DETAILED_CAPTION>"]
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