Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import torch
 from diffusers import UnCLIPPipeline
 
-pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha") # , torch_dtype=torch.float16
+pipe = pipe.to("cpu")
 
 def run(prompt):
     images = pipe(prompt, num_images_per_prompt=6).images