ex
Browse files- app.py +8 -2
- requirements.txt +2 -1
- 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,
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
-
|
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 |
-
"
|
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 |
-
|
|
|
130 |
],
|
131 |
[
|
132 |
-
"
|
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 |
]
|