Update app.py

app.py

@@ -45,7 +45,7 @@ def generate_image(prompt, negative_prompt=None, height=512, width=512, model="s
         return None
 
 # Function to refine an image using the refiner model
-def refine_image(image, prompt, negative_prompt=None, model="refiners/sd15.juggernaut.reborn.unet", num_inference_steps=15, guidance_scale=7.5):
+def refine_image(image, prompt, negative_prompt=None, model="stabilityai/stable-diffusion-xl-refiner-1.0", num_inference_steps=50, guidance_scale=7.5):
     try:
         # Use Hugging Face's image-to-image API to refine the image
         refined_image = client.image_to_image(
@@ -73,7 +73,7 @@ def generate_api():
     num_inference_steps = data.get('num_inference_steps', 50)  # Default number of inference steps
     guidance_scale = data.get('guidance_scale', 7.5)  # Default guidance scale
     model_name = data.get('model', 'stabilityai/sd-3.5')  # Base model
-    refiner_model_name = 'refiners/sd15.juggernaut.reborn.unet'  # Refiner model
+    refiner_model_name = 'stabilityai/sd-xl-refiner-1.0'  # Refiner model
     seed = data.get('seed', None)  # Seed for reproducibility, default is None
 
     if not prompt: