Update app.py

app.py

@@ -6,7 +6,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to(device).eval()
 florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True)
 
-def generate_caption():
+def generate_caption(image):
     if not isinstance(image, Image.Image):
         image = Image.fromarray(image)