Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 from threading import RLock
 from datetime import datetime
 
-preSetPrompt = "a cute tall slender athletic young nude caucasian woman. gorgeous face. relaxed smile. perfect boobs. outie pussy. pussy juice. artistic. photorealistic. cinematic. explicit. f1.4"
+preSetPrompt = "a cute tall slender athletic young nude caucasian woman. gorgeous face. sensual expression. provocative pose. perfect boobs. outie pussy. pussy juice. artistic. photorealistic. cinematic. explicit. f1.4"
 negPreSetPrompt = "[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness"
 
 lock = RLock()
@@ -16,7 +16,6 @@ def get_current_time():
     now = datetime.now()
     now2 = now
     current_time = now2.strftime("%y-%m-%d %H:%M:%S")
-    # ki = f'{kii} {current_time}'
     return current_time
 
 def load_fn(models):
@@ -37,7 +36,7 @@ load_fn(models)
 
 num_models = 6
 max_images = 6
-inference_timeout = 300
+inference_timeout = 400
 default_models = models[:num_models]
 MAX_SEED = 2**32-1
 loopcounter = 1
@@ -68,8 +67,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
     if cfg > 0: cfg = kwargs["guidance_scale"] = cfg
     if seed == -1: kwargs["seed"] = randomize_seed()
     else: kwargs["seed"] = seed
-    task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn,
-                               prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
+    task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
     await asyncio.sleep(0)
     try:
         result = await asyncio.wait_for(task, timeout=timeout)
@@ -88,7 +86,9 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
         with lock:
             # png_path = "img.png"
             global loopcounter 
-            png_path = get_current_time() + "_" + str(loopcounter) + ".png"
+            # png_path = get_current_time() + "_" + str(loopcounter) + ".png"
+            # png_path = get_current_time() + "_" + str(seed) + ".png"
+            png_path = get_current_time() + "_" + models_load[model_str] + "_" + str(seed) + ".png"
             loopcounter += 1
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
         return image