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 |
]
|