Update app.py

app.py

@@ -10,8 +10,6 @@ import cnnrnn
 from build_vocab import Vocabulary
 
 
-    
-
 # Caption generation functions
 def generate_caption_clipgpt(image, max_tokens, temperature):
     caption = clipGPT.generate_caption_clipgpt(image, max_tokens, temperature)
@@ -35,12 +33,10 @@ def generate_caption_cnnrnn(image):
 
 
 with gr.Row():
-     
     image = gr.Image(label="Upload Chest X-ray", type="pil")   
                      
        
 with gr.Row():
- 
     with gr.Column(): # Column for dropdowns and model choice
         max_tokens = gr.Dropdown(list(range(50, 101)), label="Max Tokens", value=75)
         temperature = gr.Slider(0.5, 0.9, step=0.1, label="Temperature", value=0.7) 
@@ -59,7 +55,7 @@ def predict(img, model_name, max_tokens, temperature):
     elif model_name == "ViT-CoAttention":
         return generate_caption_vitCoAtt(img)
     elif model_name == "Baseline Model CNN-RNN":
-        print(img)
+        print(img.name)
         return generate_caption_cnnrnn(img)
     else:
         return "Caption generation for this model is not yet implemented."   
@@ -68,6 +64,7 @@ def predict(img, model_name, max_tokens, temperature):
 
 examples = [[f"example{i}.jpg"] for i in range(1,4)]
 
+
 description= "You can generate captions by uploading an X-Ray and selecting a model of your choice below. Please select the number of Max Tokens and Temperature setting, if you are testing CLIP GPT2 and VIT GPT2 Models"
 title = "MedViT: A Vision Transformer-Driven Method for Generating Medical Reports 🏥🤖"