Update app.py

app.py

@@ -6,7 +6,7 @@ import threading
 import time
 from collections import OrderedDict
 import os
-from PIL import Image  # 🔥 NEW
+from PIL import Image  # 🔥 ДОБАВЛЕНО
 
 app = FastAPI()
 
@@ -16,7 +16,7 @@ app = FastAPI()
 torch.set_num_threads(2)
 
 # =========================
-# 🔥 MODEL (ТВОЙ РАБОЧИЙ)
+# 🔥 MODEL
 # =========================
 model_name = "Lykon/dreamshaper-7"
 
@@ -28,10 +28,6 @@ pipe = StableDiffusionPipeline.from_pretrained(
 
 pipe = pipe.to("cpu")
 
-# 🔥 LCM LoRA (оставляем как у тебя)
-pipe.load_lora_weights("latent-consistency/lcm-lora-sdv1-5")
-pipe.fuse_lora()
-
 pipe.enable_attention_slicing()
 
 # =========================
@@ -42,11 +38,13 @@ queue = []
 progress_db = {}
 
 MAX_HISTORY = 40
+NUM_WORKERS = 1
+
 IMG_DIR = "images"
 os.makedirs(IMG_DIR, exist_ok=True)
 
 # =========================
-# ✂️ SPLIT FUNCTION (12 частей)
+# ✂️ 12 SPLIT FUNCTION (NEW)
 # =========================
 def split_image_into_12(img_path: str):
 
@@ -87,29 +85,28 @@ def generate_ai_stream(message: str, mode="fast"):
 
     try:
         start = time.time()
-        progress_db[message] = 0
 
-        # ⚡ FAST / QUALITY
+        # ⚡ режимы
         if mode == "fast":
-            steps = 4
+            steps = 2
             cfg = 1.5
-            h, w = 256, 256
         else:
-            steps = 12
-            cfg = 6.5
-            h, w = 512, 512
+            steps = 6
+            cfg = 3.0
+
+        progress_db[message] = 0
 
-        # 🔥 fake progress
+        # 🔥 fake-progress
         for i in range(steps):
             progress_db[message] = int((i / steps) * 100)
-            time.sleep(0.08)
+            time.sleep(0.12)
 
         image = pipe(
             message,
             num_inference_steps=steps,
             guidance_scale=cfg,
-            height=h,
-            width=w
+            height=256,
+            width=256
         ).images[0]
 
         filename = f"{IMG_DIR}/img_{int(time.time()*1000)}.png"
@@ -161,10 +158,9 @@ threading.Thread(target=worker, daemon=True).start()
 
 @app.get("/")
 async def root():
-    return PlainTextResponse("⚡ Dreamshaper + LCM + FAST/QUALITY + SPLIT 12 READY")
+    return PlainTextResponse("⚡ CPU LCM Image Generator Running")
 
 
-# 🚀 FAST MODE
 @app.get("/fast")
 async def fast(message: str):
 
@@ -178,7 +174,6 @@ async def fast(message: str):
     return PlainTextResponse("accepted")
 
 
-# 🎨 QUALITY MODE
 @app.get("/quality")
 async def quality(message: str):
 
@@ -192,7 +187,6 @@ async def quality(message: str):
     return PlainTextResponse("accepted")
 
 
-# 📡 GET + PROGRESS
 @app.get("/get")
 async def get(message: str):
 
@@ -208,7 +202,6 @@ async def get(message: str):
     return PlainTextResponse(data["reply"])
 
 
-# 🖼 FILE SERVE
 @app.get("/image")
 async def get_image(path: str):