Jyothirmai commited on
Commit
3720a32
·
verified ·
1 Parent(s): fdfdf8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def load_model_1(): # CLIP-GPT2
13
 
14
  # Caption generation functions
15
  def generate_caption_clipgpt(image):
16
- image = io.imread(img)
17
  pil_image = PIL.Image.fromarray(image)
18
  caption = clipGPT.generate_caption_clipgpt(pil_image)
19
  return caption
 
13
 
14
  # Caption generation functions
15
  def generate_caption_clipgpt(image):
16
+ image = io.imread(image)
17
  pil_image = PIL.Image.fromarray(image)
18
  caption = clipGPT.generate_caption_clipgpt(pil_image)
19
  return caption