Jyothirmai
commited on
Commit
•
b51c75c
1
Parent(s):
cf05f8b
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ with gr.Blocks() as demo:
|
|
48 |
def predict(img, model_name):
|
49 |
if model_name == "CLIP-GPT2":
|
50 |
return generate_caption_clipgpt(img)
|
51 |
-
elif model_name == "ViT-GPT2"
|
52 |
return generate_caption_vitgpt(img)
|
53 |
else:
|
54 |
return "Caption generation for this model is not yet implemented."
|
|
|
48 |
def predict(img, model_name):
|
49 |
if model_name == "CLIP-GPT2":
|
50 |
return generate_caption_clipgpt(img)
|
51 |
+
elif model_name == "ViT-GPT2":
|
52 |
return generate_caption_vitgpt(img)
|
53 |
else:
|
54 |
return "Caption generation for this model is not yet implemented."
|