Update optimized.py

optimized.py

@@ -84,6 +84,7 @@ print(f"VRAM used: {torch.cuda.memory_allocated()/1e9:.2f}GB")
 @spaces.GPU
 def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale):
     # Clean up input handling
+    control_image = load_image(control_image)
     w, h = control_image.size
     scale = min(scale, 2.0)  # Cap scale factor