Update app.py
Browse files
app.py
CHANGED
|
@@ -45,6 +45,7 @@ with gr.Blocks() as demo:
|
|
| 45 |
|
| 46 |
def predict(img, model_name):
|
| 47 |
if model_name == "CLIP-GPT2":
|
|
|
|
| 48 |
return generate_caption_clipgpt(img)
|
| 49 |
# Add elif blocks for "ViT-GPT2", "ViT-CoAttention" as you implement them
|
| 50 |
else:
|
|
|
|
| 45 |
|
| 46 |
def predict(img, model_name):
|
| 47 |
if model_name == "CLIP-GPT2":
|
| 48 |
+
print(img)
|
| 49 |
return generate_caption_clipgpt(img)
|
| 50 |
# Add elif blocks for "ViT-GPT2", "ViT-CoAttention" as you implement them
|
| 51 |
else:
|