Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 4

Commit

05fd4a4

•

1 Parent(s): 299f4de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,7 @@ with gr.Blocks() as demo:
     caption = gr.Textbox(label="Generated Caption")
-    def predict(img, model_name):
         if model_name == "CLIP-GPT2":
             return generate_caption_clipgpt(img)
         elif model_name == "ViT-GPT2":

     caption = gr.Textbox(label="Generated Caption")
+    def predict(img, model_name, type="pil"):
         if model_name == "CLIP-GPT2":
             return generate_caption_clipgpt(img)
         elif model_name == "ViT-GPT2":