LPX55 commited on
Commit
f271448
·
verified ·
1 Parent(s): 88b51e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,6 +8,7 @@ import spaces
8
  import torch
9
  from diffusers import FluxControlNetModel, FluxPipeline
10
  from diffusers.pipelines import FluxControlNetPipeline
 
11
  from gradio_imageslider import ImageSlider
12
  from PIL import Image
13
  from huggingface_hub import snapshot_download, hf_hub_download
@@ -96,7 +97,7 @@ def infer(
96
  ):
97
  if randomize_seed:
98
  seed = random.randint(0, MAX_SEED)
99
- true_input_image = input_image
100
  input_image, w_original, h_original, was_resized = process_input(
101
  input_image, upscale_factor
102
  )
 
8
  import torch
9
  from diffusers import FluxControlNetModel, FluxPipeline
10
  from diffusers.pipelines import FluxControlNetPipeline
11
+ from diffusers.utils import load_image
12
  from gradio_imageslider import ImageSlider
13
  from PIL import Image
14
  from huggingface_hub import snapshot_download, hf_hub_download
 
97
  ):
98
  if randomize_seed:
99
  seed = random.randint(0, MAX_SEED)
100
+ true_input_image = load_image(input_image)
101
  input_image, w_original, h_original, was_resized = process_input(
102
  input_image, upscale_factor
103
  )