Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from share_btn import community_icon_html, loading_icon_html, share_js
|
|
11 |
|
12 |
model_id = "riffusion/riffusion-model-v1"
|
13 |
blip_model_id = "Salesforce/blip-image-captioning-base"
|
14 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id
|
15 |
# pipe = pipe.to("cuda")
|
16 |
|
17 |
blip_model = BlipForConditionalGeneration.from_pretrained(blip_model_id)
|
|
|
11 |
|
12 |
model_id = "riffusion/riffusion-model-v1"
|
13 |
blip_model_id = "Salesforce/blip-image-captioning-base"
|
14 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
15 |
# pipe = pipe.to("cuda")
|
16 |
|
17 |
blip_model = BlipForConditionalGeneration.from_pretrained(blip_model_id)
|