Update handler.py

handler.py

@@ -269,12 +269,13 @@ class EndpointHandler:
         input_image = inputs.get("image")
         
         params = data.get("parameters", dict())
-        
-        if not input_prompt:
-            raise ValueError("The prompt should not be empty")
-                        
-        logger.info(f"Prompt: {input_prompt}")
-                        
+
+         if not input_image and not input_prompt:
+            raise ValueError("Either prompt or image must be provided")
+      
+        if input_prompt:
+            logger.info(f"Prompt: {input_prompt}")
+                   
         logger.info(f"Raw parameters:")
         pprint.pprint(params)