Jyothirmai
commited on
Commit
•
8b643e1
1
Parent(s):
d2177c5
Update app.py
Browse files
app.py
CHANGED
@@ -74,7 +74,6 @@ def predict(img, model_name, max_tokens, temperature, imgID):
|
|
74 |
return generate_caption_vitCoAtt(img), getCaption(imgID)
|
75 |
elif model_name == "Baseline Model CNN-RNN":
|
76 |
img = getImageID(imgID)
|
77 |
-
print(img)
|
78 |
return generate_caption_cnnrnn(img), getCaption(imgID)
|
79 |
else:
|
80 |
return "select a model","select an image"
|
|
|
74 |
return generate_caption_vitCoAtt(img), getCaption(imgID)
|
75 |
elif model_name == "Baseline Model CNN-RNN":
|
76 |
img = getImageID(imgID)
|
|
|
77 |
return generate_caption_cnnrnn(img), getCaption(imgID)
|
78 |
else:
|
79 |
return "select a model","select an image"
|