Hugging Face's logo

Spaces:
Jyothirmai
/
image-captioning-chest-xrays
Sleeping

App Files Community
1
Jyothirmai commited on
Commit
05fd4a4
·
verified ·
1 Parent(s): 299f4de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ with gr.Blocks() as demo:
50
 
51
  caption = gr.Textbox(label="Generated Caption")
52
 
53
- def predict(img, model_name):
54
  if model_name == "CLIP-GPT2":
55
  return generate_caption_clipgpt(img)
56
  elif model_name == "ViT-GPT2":
 
50
 
51
  caption = gr.Textbox(label="Generated Caption")
52
 
53
+ def predict(img, model_name, type="pil"):
54
  if model_name == "CLIP-GPT2":
55
  return generate_caption_clipgpt(img)
56
  elif model_name == "ViT-GPT2":