Update app.py

app.py

@@ -104,12 +104,13 @@ def generate_image(data: dict):
         style_name = data['style_name']
         guidance_scale = data['guidance_scale']
         image_urls = infer(prompt, style_name=style_name, guidance_scale=guidance_scale)
-            
-            # Convert the first generated image to base64
+        
+        # Convert the first generated image to base64
         image_b64 = image_urls[0].split(",")[1]
         return {"image_base64": image_b64}
     else:
         raise HTTPException(status_code=400, detail="Missing required parameters")
+