radames HF staff commited on
Commit
f862848
1 Parent(s): 409d4bc
Files changed (3) hide show
  1. app.py +8 -2
  2. requirements.txt +2 -1
  3. utils.py +10 -4
app.py CHANGED
@@ -192,7 +192,13 @@ def flatten(l: List[List[any]]) -> List[any]:
192
 
193
 
194
  def predict_examples(
195
- prompt, negative_prompt, input_image=None, remove_bg=False, cond_mode=None, seed=-1
 
 
 
 
 
 
196
  ):
197
  print(
198
  "RUUNING EXAMPLES", prompt, negative_prompt, input_image, remove_bg, cond_mode
@@ -207,7 +213,7 @@ def predict_examples(
207
  "dpmpp_2m_sde_gpu",
208
  "karras",
209
  30,
210
- 5,
211
  1.0,
212
  )
213
 
 
192
 
193
 
194
  def predict_examples(
195
+ prompt,
196
+ negative_prompt,
197
+ input_image=None,
198
+ remove_bg=False,
199
+ cond_mode=None,
200
+ seed=-1,
201
+ cfg=10,
202
  ):
203
  print(
204
  "RUUNING EXAMPLES", prompt, negative_prompt, input_image, remove_bg, cond_mode
 
213
  "dpmpp_2m_sde_gpu",
214
  "karras",
215
  30,
216
+ cfg,
217
  1.0,
218
  )
219
 
requirements.txt CHANGED
@@ -17,4 +17,5 @@ tqdm
17
  psutil
18
  kornia>=0.7.1
19
  diffusers>=0.25.0
20
- opencv-python
 
 
17
  psutil
18
  kornia>=0.7.1
19
  diffusers>=0.25.0
20
+ opencv-python
21
+ hf_transfer
utils.py CHANGED
@@ -113,28 +113,31 @@ def downloadModels():
113
 
114
  examples = [
115
  [
116
- "An old men sit on a chair looking at the sky",
117
  "ugly distorted image, low quality, text, bad, not good ,watermark",
118
  None,
119
  False,
120
  None,
121
  1231231,
 
122
  ],
123
  [
124
- "A beautiful toucan bird flying in the sky",
125
  "ugly distorted image, low quality, text, bad, not good ,watermark",
126
  "./examples/bg.png",
127
  False,
128
  "SDXL, Background",
129
- 123123,
 
130
  ],
131
  [
132
- "graphic design label of a wine bottle detail, high quality",
133
  "ugly distorted image, low quality, text, bad, not good ,watermark",
134
  "./examples/lecun.png",
135
  True,
136
  "SDXL, Foreground",
137
  123123,
 
138
  ],
139
  [
140
  "An image of a galaxy",
@@ -143,6 +146,7 @@ examples = [
143
  True,
144
  "SDXL, Foreground",
145
  123123,
 
146
  ],
147
  [
148
  "a men jumping on swiming pool full of people",
@@ -151,6 +155,7 @@ examples = [
151
  False,
152
  "SDXL, Foreground",
153
  5350795678007195000,
 
154
  ],
155
  [
156
  "a cute cat flying over Manhattan time square",
@@ -159,5 +164,6 @@ examples = [
159
  True,
160
  "SDXL, Foreground",
161
  123123,
 
162
  ],
163
  ]
 
113
 
114
  examples = [
115
  [
116
+ "A very cute monster cat on a glass bottle",
117
  "ugly distorted image, low quality, text, bad, not good ,watermark",
118
  None,
119
  False,
120
  None,
121
  1231231,
122
+ 5,
123
  ],
124
  [
125
+ "A picture from above captures a beautiful, small toucan bird flying in the sky.",
126
  "ugly distorted image, low quality, text, bad, not good ,watermark",
127
  "./examples/bg.png",
128
  False,
129
  "SDXL, Background",
130
+ 1234144,
131
+ 8,
132
  ],
133
  [
134
+ "a photo a men surrounded by a crowd of people in a circle",
135
  "ugly distorted image, low quality, text, bad, not good ,watermark",
136
  "./examples/lecun.png",
137
  True,
138
  "SDXL, Foreground",
139
  123123,
140
+ 10,
141
  ],
142
  [
143
  "An image of a galaxy",
 
146
  True,
147
  "SDXL, Foreground",
148
  123123,
149
+ 10,
150
  ],
151
  [
152
  "a men jumping on swiming pool full of people",
 
155
  False,
156
  "SDXL, Foreground",
157
  5350795678007195000,
158
+ 10,
159
  ],
160
  [
161
  "a cute cat flying over Manhattan time square",
 
164
  True,
165
  "SDXL, Foreground",
166
  123123,
167
+ 10,
168
  ],
169
  ]