Spaces:

SnehaTiwari
/

Fashion-Image-generation

Runtime error

SnehaTiwari commited on Jul 13, 2023

Commit

54ea64b

1 Parent(s): fd061e3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import math
 import gradio as gr
 pipeline = StableDiffusionPipeline.from_pretrained("MohamedRashad/diffusion_fashion", torch_dtype=torch.float16)
 def image_grid(imgs, rows, cols):
     assert len(imgs) == rows*cols

 import gradio as gr
 pipeline = StableDiffusionPipeline.from_pretrained("MohamedRashad/diffusion_fashion", torch_dtype=torch.float16)
+pipeline.to("cuda")
 def image_grid(imgs, rows, cols):
     assert len(imgs) == rows*cols