Spaces:

nroggendorff
/

cats

Runtime error

nroggendorff commited on Apr 4, 2024

Commit

a2809d8

verified ·

1 Parent(s): 772093a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from diffusers import DiffusionPipeline
 def generate_image():
-    pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
     pipe = pipeline.to("cuda")
     image = pipe().images[0]
     return image

 from diffusers import DiffusionPipeline
 def generate_image():
+    pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats", use_safetensors=True)
     pipe = pipeline.to("cuda")
     image = pipe().images[0]
     return image