Muzmmillcoste commited on
Commit
c202d55
·
1 Parent(s): cf691fd

change captioner funcion little bit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def image_to_base64_str(pil_image):
39
  def captioner(image):
40
  base64_image = image_to_base64_str(image)
41
  result = get_completion(base64_image)
42
- return result
43
 
44
  # Interface
45
  gr.close_all()
 
39
  def captioner(image):
40
  base64_image = image_to_base64_str(image)
41
  result = get_completion(base64_image)
42
+ return result[0]['generated_text']
43
 
44
  # Interface
45
  gr.close_all()