Update app.py

app.py

@@ -40,7 +40,7 @@ with gr.Blocks() as demo:
             ]
 
         
-        image = gr.Image(label="Upload Chest X-ray")    
+        image = gr.Image(label="Upload Chest X-ray", type="pil")    
         
         sample_images_gallery = gr.Gallery(value = sample_images,label="Sample Images")
     
@@ -50,7 +50,7 @@ with gr.Blocks() as demo:
     
     caption = gr.Textbox(label="Generated Caption") 
 
-    def predict(img, model_name, type="pil"):
+    def predict(img, model_name):
         if model_name == "CLIP-GPT2":
             return generate_caption_clipgpt(img)
         elif model_name == "ViT-GPT2":