gokaygokay commited on
Commit
c6b2d4a
1 Parent(s): e4c639b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -177,6 +177,9 @@ def gradio_process_image(input_image, resolution, num_inference_steps, strength,
177
  print("Starting image processing...")
178
  torch.cuda.empty_cache()
179
 
 
 
 
180
  condition_image = prepare_image(input_image, resolution, hdr)
181
 
182
  prompt = "masterpiece, best quality, highres"
@@ -203,7 +206,7 @@ def gradio_process_image(input_image, resolution, num_inference_steps, strength,
203
  input_array = np.array(input_image)
204
  result_array = np.array(result)
205
 
206
- return [input_array, result_array]
207
 
208
  title = """<h1 align="center">Image Upscaler with Tile Controlnet</h1>
209
  <p align="center">The main ideas come from</p>
 
177
  print("Starting image processing...")
178
  torch.cuda.empty_cache()
179
 
180
+ # Update the output slider with a processing message
181
+ yield gr.update(value="Currently Processing the Image...")
182
+
183
  condition_image = prepare_image(input_image, resolution, hdr)
184
 
185
  prompt = "masterpiece, best quality, highres"
 
206
  input_array = np.array(input_image)
207
  result_array = np.array(result)
208
 
209
+ yield [input_array, result_array]
210
 
211
  title = """<h1 align="center">Image Upscaler with Tile Controlnet</h1>
212
  <p align="center">The main ideas come from</p>