Jyothirmai commited on
Commit
8b643e1
1 Parent(s): d2177c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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"