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"
|