Update app.py

app.py

@@ -36,28 +36,31 @@ Your output must be: a single clean answer string only.
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+from smolagents.schema import Tool
+
 class ImageCaptioningTool(Tool):
     name = "image_captioner"
-    description = "Generate a caption for an image."
+    description = "Generate a caption for an image using a prompt or question."
+
     inputs = {
         "image": {
-            "type": "dict",
-            "description": "A dictionary containing image data in PIL format."
+            "type": "image",
+            "description": "An image file input."
         },
         "question": {
-            "type": "str",
-            "description": "A question or prompt about the image."
+            "type": "string",
+            "description": "A prompt or question about the image."
         }
     }
-    output_type = "text"
+
+    output_type = "string"
 
     def run(self, inputs: dict) -> str:
-        image = inputs.get("image")
-        if not image:
-            return "No image provided."
-        # Here you can process the image using your model
-        return "This is a placeholder caption for the uploaded image."
-        
+        image = inputs["image"]
+        question = inputs["question"]
+        # Dummy logic – replace with actual model
+        return f"Captioning image for: {question}"
+       
 # class ImageCaptioningTool(Tool):
 #     name = "image_captioner"
 #     description = "Generate a caption for an image."