Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def load_translation_models(translation_model_id):
|
|
26 |
|
27 |
def pipeline_generate(diffusion_model_id):
|
28 |
pipe = StableDiffusionPipeline.from_pretrained(
|
29 |
-
diffusion_model_id,
|
30 |
use_auth_token=HUGGING_FACE_TOKEN
|
31 |
)
|
32 |
pipe = pipe.to(DEVICE_NAME)
|
|
|
26 |
|
27 |
def pipeline_generate(diffusion_model_id):
|
28 |
pipe = StableDiffusionPipeline.from_pretrained(
|
29 |
+
diffusion_model_id,
|
30 |
use_auth_token=HUGGING_FACE_TOKEN
|
31 |
)
|
32 |
pipe = pipe.to(DEVICE_NAME)
|