Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def generate_caption_vitgpt(image):
|
|
20 |
return caption
|
21 |
|
22 |
def generate_caption_vitCoAtt(image):
|
23 |
-
caption =
|
24 |
return caption
|
25 |
|
26 |
|
|
|
20 |
return caption
|
21 |
|
22 |
def generate_caption_vitCoAtt(image):
|
23 |
+
caption = ViTCoAtt.CaptionSampler.main(image)
|
24 |
return caption
|
25 |
|
26 |
|