Jyothirmai commited on
Commit
b51c75c
1 Parent(s): cf05f8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ with gr.Blocks() as demo:
48
  def predict(img, model_name):
49
  if model_name == "CLIP-GPT2":
50
  return generate_caption_clipgpt(img)
51
- elif model_name == "ViT-GPT2"
52
  return generate_caption_vitgpt(img)
53
  else:
54
  return "Caption generation for this model is not yet implemented."
 
48
  def predict(img, model_name):
49
  if model_name == "CLIP-GPT2":
50
  return generate_caption_clipgpt(img)
51
+ elif model_name == "ViT-GPT2":
52
  return generate_caption_vitgpt(img)
53
  else:
54
  return "Caption generation for this model is not yet implemented."