Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -262,6 +262,7 @@ def get_random_model(models):
|
|
262 |
global last_two_models
|
263 |
model = None
|
264 |
priorities = [
|
|
|
265 |
"kandinsky-community",
|
266 |
"Kolors-diffusers",
|
267 |
"Juggernaut",
|
@@ -282,7 +283,7 @@ def get_random_model(models):
|
|
282 |
print("Choosing randomly")
|
283 |
model = random.choice(models)
|
284 |
last_two_models.append(model)
|
285 |
-
last_two_models = last_two_models[-
|
286 |
return model
|
287 |
|
288 |
def nsfw_check(attempts = 1):
|
|
|
262 |
global last_two_models
|
263 |
model = None
|
264 |
priorities = [
|
265 |
+
"FLUX",
|
266 |
"kandinsky-community",
|
267 |
"Kolors-diffusers",
|
268 |
"Juggernaut",
|
|
|
283 |
print("Choosing randomly")
|
284 |
model = random.choice(models)
|
285 |
last_two_models.append(model)
|
286 |
+
last_two_models = last_two_models[-3:]
|
287 |
return model
|
288 |
|
289 |
def nsfw_check(attempts = 1):
|