Gainward777 commited on
Commit
9d6c0b1
·
verified ·
1 Parent(s): f13d198

Update sd/zerogpu_controller.py

Browse files
Files changed (1) hide show
  1. sd/zerogpu_controller.py +1 -6
sd/zerogpu_controller.py CHANGED
@@ -1,4 +1,3 @@
1
- #from ui.gradio_ui import ui
2
  import spaces
3
 
4
  from sd.prompt_helper import helper
@@ -6,9 +5,7 @@ from PIL import Image
6
 
7
  from sd.utils.utils import *
8
  from utils.utils import sketch_process, prompt_preprocess
9
- #from sd.sd_controller import Controller
10
 
11
- #controller=Controller()
12
 
13
  MODELS_NAMES=["cagliostrolab/animagine-xl-3.1",
14
  "stabilityai/stable-diffusion-xl-base-1.0"]
@@ -83,6 +80,4 @@ def get_second_result(img, prompt, negative_prompt,
83
 
84
 
85
  def get_help_w_prompt(img):
86
- return helper.get_help(img["composite"]) #Image.fromarray(img["composite"], "RGB"))
87
-
88
- #ui(get_first_result, get_second_result) #controller)
 
 
1
  import spaces
2
 
3
  from sd.prompt_helper import helper
 
5
 
6
  from sd.utils.utils import *
7
  from utils.utils import sketch_process, prompt_preprocess
 
8
 
 
9
 
10
  MODELS_NAMES=["cagliostrolab/animagine-xl-3.1",
11
  "stabilityai/stable-diffusion-xl-base-1.0"]
 
80
 
81
 
82
  def get_help_w_prompt(img):
83
+ return helper.get_help(img["composite"])