Update app.py

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 import numpy as np
 import random
 
-# import spaces #[uncomment to use ZeroGPU]
+import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 
 MODEL_OPTIONS = [
     ("stabilityai/sdxl-turbo", "SDXL Turbo (Быстро)"),
@@ -31,13 +30,13 @@ def load_pipelines():
     # SDXL Turbo
     mid = "stabilityai/sdxl-turbo"
     pipe = DiffusionPipeline.from_pretrained(mid, torch_dtype=torch_dtype)
-    pipe = pipe.to(device)
+    # pipe = pipe.to(device)
     PIPELINES[mid] = pipe
 
     # SD v1-4
     mid = "CompVis/stable-diffusion-v1-4"
     pipe = DiffusionPipeline.from_pretrained(mid, torch_dtype=torch_dtype)
-    pipe = pipe.to(device)
+    # pipe = pipe.to(device)
     PIPELINES[mid] = pipe
 
     # # Qwen-Image 
@@ -51,9 +50,9 @@ load_pipelines()
 
 
 
-# @spaces.GPU #[uncomment to use ZeroGPU]
+@spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
-    model_id,  # Добавил
+    model_id, 
     prompt,
     negative_prompt,
     seed,
@@ -69,7 +68,8 @@ def infer(
 
     generator = torch.Generator().manual_seed(seed)
     
-    pipe = PIPELINES[model_id]  
+    pipe = PIPELINES[model_id] 
+    pipe = pipe.to(device)
 
     image = pipe(
         prompt=prompt,