Update app.py

app.py

@@ -47,7 +47,7 @@ with gr.Row():
     
 caption = gr.Textbox(label="Generated Caption") 
     
-def predict(img, model_name, max_tokens, temperature):
+def predict(img, model_name, max_tokens, temperature, examples):
     if model_name == "CLIP-GPT2":
         return generate_caption_clipgpt(img, max_tokens, temperature)
     elif model_name == "ViT-GPT2":
@@ -55,7 +55,6 @@ 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.name)
         return generate_caption_cnnrnn(img)
     else:
         return "Caption generation for this model is not yet implemented."