Lisandro commited on
Commit
910daa5
1 Parent(s): 78cb566

feat: Refactor app.py to improve code structure and add image upload functionality

Browse files
Files changed (1) hide show
  1. app.py +12 -4
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 gradio_process_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,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=gradio_process_image,
51
- inputs=[input_image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name],
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()