Update app.py
Browse files
app.py
CHANGED
|
@@ -13,8 +13,6 @@ def load_model_1(): # CLIP-GPT2
|
|
| 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
|
| 20 |
|
|
|
|
| 13 |
|
| 14 |
# Caption generation functions
|
| 15 |
def generate_caption_clipgpt(image):
|
|
|
|
|
|
|
| 16 |
caption = clipGPT.generate_caption_clipgpt(pil_image)
|
| 17 |
return caption
|
| 18 |
|