Spaces:
Sleeping
Sleeping
feat: Refactor app.py to improve code structure and add image upload functionality
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ tile_upscaler_url = "https://gokaygokay-tileupscalerv2.hf.space"
|
|
7 |
|
8 |
client_tile_upscaler = Client(tile_upscaler_url)
|
9 |
|
10 |
-
def
|
11 |
result = client_tile_upscaler.predict(
|
12 |
param_0=handle_file(image),
|
13 |
param_1=resolution,
|
@@ -19,8 +19,12 @@ def gradio_process_image(image, resolution, num_inference_steps, strength, hdr,
|
|
19 |
param_7=scheduler_name,
|
20 |
api_name="/wrapper"
|
21 |
)
|
22 |
-
print(result)
|
23 |
return result
|
|
|
|
|
|
|
|
|
|
|
24 |
|
25 |
|
26 |
with gr.Blocks() as demo:
|
@@ -47,9 +51,13 @@ with gr.Blocks() as demo:
|
|
47 |
|
48 |
|
49 |
run_button.click(
|
50 |
-
fn=
|
51 |
-
inputs=[
|
52 |
outputs=output_slider
|
|
|
|
|
|
|
|
|
53 |
)
|
54 |
|
55 |
demo.launch()
|
|
|
7 |
|
8 |
client_tile_upscaler = Client(tile_upscaler_url)
|
9 |
|
10 |
+
def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name):
|
11 |
result = client_tile_upscaler.predict(
|
12 |
param_0=handle_file(image),
|
13 |
param_1=resolution,
|
|
|
19 |
param_7=scheduler_name,
|
20 |
api_name="/wrapper"
|
21 |
)
|
|
|
22 |
return result
|
23 |
+
|
24 |
+
def clear_output(image_slider):
|
25 |
+
image_slider[0] = None
|
26 |
+
image_slider[1] = None
|
27 |
+
return image_slider
|
28 |
|
29 |
|
30 |
with gr.Blocks() as demo:
|
|
|
51 |
|
52 |
|
53 |
run_button.click(
|
54 |
+
fn=clear_output,
|
55 |
+
inputs=[output_slider]
|
56 |
outputs=output_slider
|
57 |
+
).then(
|
58 |
+
upscale_image,
|
59 |
+
[input_image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name],
|
60 |
+
output_slider
|
61 |
)
|
62 |
|
63 |
demo.launch()
|