Update app.py

app.py

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