Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -13,8 +13,8 @@ sd_2_0_2x = StableDiffusionLatentUpscalePipelin.from_pretrained(model_2x, torch_
|
|
13 |
sd_2_1_4x = StableDiffusionUpscalePipeline.from_pretrained(model_4x, torch_dtype=torch.float16, revision="fp16") if torch.cuda.is_available() else StableDiffusionUpscalePipeline.from_pretrained(model_4x)
|
14 |
|
15 |
# Define the input and output components for the Gradio interface
|
16 |
-
input_image = gr.inputs.Image()
|
17 |
-
output_image = gr.outputs.Image()
|
18 |
|
19 |
# Define the function that will be called when the interface is used
|
20 |
def upscale_image(model, image):
|
|
|
13 |
sd_2_1_4x = StableDiffusionUpscalePipeline.from_pretrained(model_4x, torch_dtype=torch.float16, revision="fp16") if torch.cuda.is_available() else StableDiffusionUpscalePipeline.from_pretrained(model_4x)
|
14 |
|
15 |
# Define the input and output components for the Gradio interface
|
16 |
+
input_image = gr.inputs.Image(type="filepath")
|
17 |
+
output_image = gr.outputs.Image(type="filepath")
|
18 |
|
19 |
# Define the function that will be called when the interface is used
|
20 |
def upscale_image(model, image):
|