Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -262,7 +262,9 @@ def get_random_model(models):
|
|
262 |
global last_two_models
|
263 |
model = None
|
264 |
priorities = [
|
265 |
-
"
|
|
|
|
|
266 |
"kandinsky-community",
|
267 |
"Kolors-diffusers",
|
268 |
"Juggernaut",
|
@@ -283,7 +285,7 @@ def get_random_model(models):
|
|
283 |
print("Choosing randomly")
|
284 |
model = random.choice(models)
|
285 |
last_two_models.append(model)
|
286 |
-
last_two_models = last_two_models[-
|
287 |
return model
|
288 |
|
289 |
def nsfw_check(attempts=1):
|
|
|
262 |
global last_two_models
|
263 |
model = None
|
264 |
priorities = [
|
265 |
+
"stabilityai/stable-diffusion-3.5-large-turbo",
|
266 |
+
"stabilityai/stable-diffusion-3.5-large",
|
267 |
+
"black-forest-labs",
|
268 |
"kandinsky-community",
|
269 |
"Kolors-diffusers",
|
270 |
"Juggernaut",
|
|
|
285 |
print("Choosing randomly")
|
286 |
model = random.choice(models)
|
287 |
last_two_models.append(model)
|
288 |
+
last_two_models = last_two_models[-5:]
|
289 |
return model
|
290 |
|
291 |
def nsfw_check(attempts=1):
|