Update app.py
Browse files
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(
|
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
|