Update main.py

main.py

@@ -19,7 +19,8 @@ except Exception as e:
 async def get_caption(image: UploadFile = File(...), context: str = Form(...)):
     try:
         contents = await image.read()
-        image_data = handle_file(contents, orig_name=image.filename)
+        # Remove orig_name if it's not supported by handle_file
+        image_data = handle_file(contents)
 
         result = client.predict(
             image=image_data,