Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from diffusers import StableDiffusionPipeline
|
|
3 |
import gradio as gr
|
4 |
|
5 |
# Use the CompVis stable-diffusion-v1-4 model
|
6 |
-
model_id = "
|
7 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) # float32 for CPU
|
8 |
pipe.to("cpu") # Ensure it runs on CPU
|
9 |
|
|
|
3 |
import gradio as gr
|
4 |
|
5 |
# Use the CompVis stable-diffusion-v1-4 model
|
6 |
+
model_id = "prompthero/openjourney"
|
7 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32) # float32 for CPU
|
8 |
pipe.to("cpu") # Ensure it runs on CPU
|
9 |
|