Geek7 commited on
Commit
535efbb
·
verified ·
1 Parent(s): fed8daa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def generate_image(prompt, negative_prompt=None, height=512, width=512, model="s
45
  return None
46
 
47
  # Function to refine an image using the refiner model
48
- def refine_image(image, prompt, negative_prompt=None, model="stabilityai/stable-diffusion-xl-refiner-1.0", num_inference_steps=50, guidance_scale=7.5):
49
  try:
50
  # Use Hugging Face's image-to-image API to refine the image
51
  refined_image = client.image_to_image(
 
45
  return None
46
 
47
  # Function to refine an image using the refiner model
48
+ def refine_image(image, prompt, negative_prompt=None, model="stabilityai/stable-diffusion-xl-refiner-1.0", num_inference_steps=15, guidance_scale=7.5):
49
  try:
50
  # Use Hugging Face's image-to-image API to refine the image
51
  refined_image = client.image_to_image(