Update app.py

app.py

@@ -36,19 +36,40 @@ Your output must be: a single clean answer string only.
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-
 class ImageCaptioningTool(Tool):
     name = "image_captioner"
     description = "Generate a caption for an image."
-    inputs = {"image": Any, "question": "str"} 
+    inputs = {
+        "image": {
+            "type": "dict",
+            "description": "A dictionary containing image data in PIL format."
+        },
+        "question": {
+            "type": "str",
+            "description": "A question or prompt about the image."
+        }
+    }
     output_type = "text"
-    
+
     def run(self, inputs: dict) -> str:
         image = inputs.get("image")
         if not image:
             return "No image provided."
-        # You could run your model here instead
+        # Here you can process the image using your model
         return "This is a placeholder caption for the uploaded image."
+        
+# class ImageCaptioningTool(Tool):
+#     name = "image_captioner"
+#     description = "Generate a caption for an image."
+#     inputs = {"image": Any, "question": "str"} 
+#     output_type = "text"
+    
+#     def run(self, inputs: dict) -> str:
+#         image = inputs.get("image")
+#         if not image:
+#             return "No image provided."
+#         # You could run your model here instead
+#         return "This is a placeholder caption for the uploaded image."
 
 
 # ---------------------- TOOL CONFIGURATION ---------------------- #