wxwtest commited on
Commit
c715c44
·
verified ·
1 Parent(s): c9d56e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -67,6 +67,9 @@ def predict(input_image_path, prompt, negative_prompt, controlnet_scale):
67
  prompt = remove_color(prompt)
68
  print(prompt)
69
 
 
 
 
70
  output_image = pipe(
71
  image=resize_base_image,
72
  control_image=resize_image,
@@ -79,6 +82,7 @@ def predict(input_image_path, prompt, negative_prompt, controlnet_scale):
79
  eta=1.0,
80
  ).images[0]
81
  print(f"Time taken: {time.time() - last_time}")
 
82
  output_image = output_image.resize(input_image.size, Image.LANCZOS)
83
  return output_image
84
 
 
67
  prompt = remove_color(prompt)
68
  print(prompt)
69
 
70
+
71
+
72
+
73
  output_image = pipe(
74
  image=resize_base_image,
75
  control_image=resize_image,
 
82
  eta=1.0,
83
  ).images[0]
84
  print(f"Time taken: {time.time() - last_time}")
85
+
86
  output_image = output_image.resize(input_image.size, Image.LANCZOS)
87
  return output_image
88