Spaces:

SnehaTiwari
/

Fashion-Image-generation

Runtime error

SnehaTiwari commited on Jul 13, 2023

Commit

3b76d9a

1 Parent(s): 54ea64b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ from PIL import Image
 import math
 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

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