minor bug fixes

app.py

@@ -43,7 +43,7 @@ class Ours:
             CLIPTextModelWithProjection.from_pretrained(
                 "laion/CLIP-ViT-bigG-14-laion2B-39B-b160k",
                 projection_dim=1280,
-                torch_dtype=torch.float16,
+                torch_dtype=torch.float32,
             )
             .eval()
             .requires_grad_(False)
@@ -55,7 +55,7 @@ class Ours:
 
         prior = PriorTransformer.from_pretrained(
             "ECLIPSE-Community/ECLIPSE_KandinskyV22_Prior",
-            torch_dtype=torch.float16,
+            torch_dtype=torch.float32,
         )
 
         self.pipe_prior = KandinskyPriorPipeline.from_pretrained(
@@ -63,11 +63,11 @@ class Ours:
             prior=prior,
             text_encoder=text_encoder,
             tokenizer=tokenizer,
-            torch_dtype=torch.float16,
+            torch_dtype=torch.float32,
         ).to(device)
 
         self.pipe = DiffusionPipeline.from_pretrained(
-            "kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16
+            "kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float32
         ).to(device)
 
     def inference(self, text, negative_text, steps, guidance_scale):

requirements.txt

@@ -9,7 +9,6 @@ torch==2.0.0
 torchvision==0.15.1
 tqdm==4.66.1
 transformers==4.34.1
-gradio
 jmespath
 opencv-python
 PyWavelet