Update multimodel_services/replicate_generation_service.py

multimodel_services/replicate_generation_service.py

@@ -35,10 +35,9 @@ def generate_image_with_model(model_name: str, prompt: str, ui_params: Optional[
             input_data["image_input"] = [image_data_url]
         
         print(f"Generating with {model_name}")
-        print(f"Input: {input_data}")
         
         # Generate image
-        output = replicate.run(model_name, input=input_data)
+        output = client.run(model_name, input=input_data)
         
         # Handle different output types
         if hasattr(output, 'read'):