Update main.py

main.py

@@ -13,8 +13,32 @@ import numpy as np
 
 processor = SegformerImageProcessor.from_pretrained("mattmdjaga/segformer_b2_clothes")
 model = TFSegformerForSemanticSegmentation.from_pretrained("mattmdjaga/segformer_b2_clothes")
-
-
+# d2c2605e-b4a5-4ab6-aa32-fcf81e6d27a2
+# 022ee431-4073-424c-8298-68cb75352785
+# e8346eb2-6187-4748-a42f-7241580ee1f1
+# ae337a18-427c-4035-a73b-a89b97a14b6c
+# c177417d-8b42-423a-8dc1-4b02c7a686ae
+# 2347e0bf-ab79-4654-817e-684f0d9e6dff
+# d01a7eb8-9f10-4abe-b142-fd5248fb7e88
+# 5e6a594a-4bc6-4b06-bed6-5ad57804e307
+# 7f6d4a3d-f3e3-4abf-ab8d-a21bb64a5ee7
+# 39110d90-70eb-4fed-bb40-3d0bf52c73bf
+# 4b35363f-a524-418b-8b2c-ee141e642a8b
+# dc80a8a4-0b98-4d54-b3e4-b7c797bc2527
+# 5113d4e5-6574-4115-bdd0-e02b5d7cfb87
+# aba3968d-b924-45d6-a98f-90ff6b12989c
+# da6053eb-c352-4374-a459-2a9a5eaaa64b
+# d46c04cb-6ce1-4657-bbfd-8c0c77d6a162
+# 06b2ac41-5dc0-461e-85d8-abbac08c5491
+# 1008a105-be30-4564-b5d2-edbd4e6a5c3d
+# 596e85a1-b3ae-49fb-ae11-cee1c4f78eee
+# 3d1ca348-aed2-4fa3-86dd-7583f2f5ef4a
+# eaca0864-70a4-4653-8dc7-f5ba3918326f
+# fcae63ee-262a-44d9-8941-cdf36ae371ad
+# 9f4b2ef5-e6b5-42a9-b459-8958402dfcc5
+# 4d012f3f-13f7-4a92-b1f6-ac0e65002803
+# 72b18186-fb64-46a4-b4f2-a91af84dd398
+# 501eba46-a956-4649-96aa-2d9cc0f048bf
 app = Flask(__name__)
 cors = CORS(app)
 app.config['CORS_HEADERS'] = 'Content-Type'
@@ -54,8 +78,8 @@ def generator():
         
         prompt ="A "+cameraAngle+" portrait of a "+age+" "+base+" with a "+expression+" and "+bodyType+" body, "+action+", "+boobs+" boobs, "+ass+" ass,  "+clothing+", "+hairStyle+" hair, "+quality+" quality, "+clothingColor+" "+clothingType+", "+clothingMaterial+" with "+clothingPattern+", "+lighting+" in "+setting+" with "+background+" in background"
         model = "dreamshaperXL10_alpha2.safetensors [c8afe2ef]"
-        if style == "photographic":
-            model = "juggernautXL_v45.safetensors [e75f5471]"
+        # if style == "photographic":
+        #     model = "juggernautXL_v45.safetensors [e75f5471]"
         # model = "sd_xl_base_1.0.safetensors [be9edd61]"
         
         payload = {