Move to CPU

app.py

@@ -3,7 +3,6 @@ from diffusers import StableDiffusionPipeline
 import torch
 
 pipe = StableDiffusionPipeline.from_pretrained("MohamedRashad/diffusion_fashion", torch_dtype=torch.float16)
-pipe.to("cuda")
 
 def generate_image(text):
     images = pipe(text).images
@@ -13,7 +12,7 @@ def generate_image(text):
 diffusion_interface = gr.Interface(
     generate_image,
     gr.Textbox(lines=1, label="Input"),
-    gr.outputs.Image(type="pil", label="Output"),
+    gr.Image(type="pil", label="Output"),
     title="Diffusion4Fashion: Generate cool clothes!",
     description="<center><p>Enter a description about a piece of cloth and the model will generate an image.</p></center>",
     examples=["A photo of a dress, made in 2019, color is Red, Casual usage, Women's cloth, something for the summer season, on white background"],