Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,16 +16,16 @@ sd_2_1_4x = StableDiffusionUpscalePipeline.from_pretrained(model_4x, torch_dtype
|
|
| 16 |
|
| 17 |
# Define the function that will be called when the interface is used
|
| 18 |
|
| 19 |
-
def upscale_image(model, input_image, prompt,
|
| 20 |
# Convert the image
|
| 21 |
generator = torch.manual_seed(999999)
|
| 22 |
input_image = Image.open(input_image).convert("RGB")
|
| 23 |
# Upscale the image using the selected model
|
| 24 |
if model == "SD 2.1 4x Upscaler":
|
| 25 |
low_res_img = input_image.resize((128, 128))
|
| 26 |
-
upscaled_image = sd_2_1_4x(prompt,
|
| 27 |
else:
|
| 28 |
-
upscaled_image = sd_2_0_2x(prompt,
|
| 29 |
# Return the upscaled image
|
| 30 |
return upscaled_image
|
| 31 |
|
|
@@ -35,7 +35,6 @@ gr.Interface(
|
|
| 35 |
inputs=[gr.Radio(["SD 2.0 2x Latent Upscaler", "SD 2.1 4x Upscaler"], label="Models:"),
|
| 36 |
gr.Image(type="filepath", label = "Raw Image"),
|
| 37 |
gr.Textbox(label='Guide the AI Upscaling'),
|
| 38 |
-
gr.Textbox(label='What you do not want AI to generate during Upscaling'),
|
| 39 |
gr.Slider(minimum=0, value=0, maximum=3, label='Guidance Scale')],
|
| 40 |
outputs=gr.Image(type="filepath", label = "Upscaled Image"),
|
| 41 |
title="SD Image Upscaler",
|
|
|
|
| 16 |
|
| 17 |
# Define the function that will be called when the interface is used
|
| 18 |
|
| 19 |
+
def upscale_image(model, input_image, prompt, guidance):
|
| 20 |
# Convert the image
|
| 21 |
generator = torch.manual_seed(999999)
|
| 22 |
input_image = Image.open(input_image).convert("RGB")
|
| 23 |
# Upscale the image using the selected model
|
| 24 |
if model == "SD 2.1 4x Upscaler":
|
| 25 |
low_res_img = input_image.resize((128, 128))
|
| 26 |
+
upscaled_image = sd_2_1_4x(prompt, image=low_res_img, num_inference_steps=5, guidance_scale=guidance).images[0]
|
| 27 |
else:
|
| 28 |
+
upscaled_image = sd_2_0_2x(prompt, image=input_image, num_inference_steps=5, guidance_scale=guidance).images[0]
|
| 29 |
# Return the upscaled image
|
| 30 |
return upscaled_image
|
| 31 |
|
|
|
|
| 35 |
inputs=[gr.Radio(["SD 2.0 2x Latent Upscaler", "SD 2.1 4x Upscaler"], label="Models:"),
|
| 36 |
gr.Image(type="filepath", label = "Raw Image"),
|
| 37 |
gr.Textbox(label='Guide the AI Upscaling'),
|
|
|
|
| 38 |
gr.Slider(minimum=0, value=0, maximum=3, label='Guidance Scale')],
|
| 39 |
outputs=gr.Image(type="filepath", label = "Upscaled Image"),
|
| 40 |
title="SD Image Upscaler",
|