Update app.py

app.py

@@ -27,7 +27,11 @@ from src.pipelines.pipeline_kandinsky_subject_prior import KandinskyPriorPipelin
 from diffusers import DiffusionPipeline
 from PIL import Image
 
-__device__ = "cuda" if torch.cuda.is_available() else "cpu"
+__device__ = "cpu"
+__dtype__ = torch.float32
+if torch.cuda.is_available():
+    __device__ = "cuda"
+    __dtype__ = torch.float16
 
 class Model:
     def __init__(self):
@@ -37,7 +41,7 @@ class Model:
             CLIPTextModelWithProjection.from_pretrained(
                 "laion/CLIP-ViT-bigG-14-laion2B-39B-b160k",
                 projection_dim=1280,
-                torch_dtype=torch.float16,
+                torch_dtype=__dtype__,
             )
             .eval()
             .requires_grad_(False)
@@ -49,17 +53,17 @@ class Model:
 
         prior = PriorTransformer.from_pretrained(
             "ECLIPSE-Community/Lambda-ECLIPSE-Prior-v1.0",
-            torch_dtype=torch.float16,
+            torch_dtype=__dtype__,
         )
 
         self.pipe_prior = KandinskyPriorPipeline.from_pretrained(
             "kandinsky-community/kandinsky-2-2-prior",
             prior=prior,
-            torch_dtype=torch.float16,
+            torch_dtype=__dtype__,
         ).to(self.device)
 
         self.pipe = DiffusionPipeline.from_pretrained(
-            "kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16
+            "kandinsky-community/kandinsky-2-2-decoder", torch_dtype=__dtype__
         ).to(self.device)
         
     def inference(self, raw_data):