Update app.py

app.py

@@ -59,8 +59,6 @@ def getCaption(imgID):
     return real_captions[imgID]
     
 def getImageID(imgID):
-    if imgID == "5" or imgID == "6":
-        return 0
     imgIDs = {"1":"/content/drive/MyDrive/cnn-rnn/NLMCXR_png/CXR412_IM-2056_0",
           "2":"/content/drive/MyDrive/cnn-rnn/NLMCXR_png/CXR545_IM-2149_0",
           "3":"/content/drive/MyDrive/cnn-rnn/NLMCXR_png/CXR849_IM-2371_0",
@@ -76,10 +74,7 @@ def predict(img, model_name, max_tokens, temperature, imgID):
         return generate_caption_vitCoAtt(img),  getCaption(imgID)
     elif model_name == "Baseline Model CNN-RNN":
         img = getImageID(imgID)
-        if img != 0:
-            return generate_caption_cnnrnn(img), getCaption(imgID)
-        else:
-            return "The images 5 and 6 are not chest xrays and the baseline model cannot be implemented try a different sample", "The images 5 and 6 are not chest xrays and the baseline model cannot be implemented try a different sample"
+        return generate_caption_cnnrnn(img), getCaption(imgID)
     else:
         return "select a model","select an image"