Update app.py

app.py

@@ -5,7 +5,6 @@ import random
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
-from PIL import Image
 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -14,7 +13,7 @@ torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 MODEL_OPTIONS = [
     ("stabilityai/sdxl-turbo", "SDXL Turbo (Быстро)"),
     ("CompVis/stable-diffusion-v1-4", "Stable Diffusion v1-4 (Классика)"),
-    ("Qwen/Qwen-Image", "Топ модель, но долго"),
+    # ("Qwen/Qwen-Image", "Топ модель, но долго"),
 ]
 DEFAULT_MODEL_ID = "stabilityai/sdxl-turbo"
 
@@ -41,11 +40,11 @@ def load_pipelines():
     pipe = pipe.to(device)
     PIPELINES[mid] = pipe
 
-    # Qwen-Image 
-    mid = "Qwen/Qwen-Image"
-    pipe = DiffusionPipeline.from_pretrained(mid, torch_dtype=torch_dtype)
-    pipe = pipe.to(device)
-    PIPELINES[mid] = pipe
+    # # Qwen-Image 
+    # mid = "Qwen/Qwen-Image"
+    # pipe = DiffusionPipeline.from_pretrained(mid, torch_dtype=torch_dtype)
+    # pipe = pipe.to(device)
+    # PIPELINES[mid] = pipe
 
 # Вызываем сразу при импорте (на сборке образа и при старте Space)
 load_pipelines()