nenene
commited on
Commit
·
554790e
1
Parent(s):
db37a47
change pipeline to cpu
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from transparent_background import Remover
|
|
| 9 |
# Initialize the Diffusion Pipeline
|
| 10 |
model_id = "yahoo-inc/photo-background-generation"
|
| 11 |
pipeline = DiffusionPipeline.from_pretrained(model_id, custom_pipeline=model_id)
|
| 12 |
-
pipeline = pipeline.to('
|
| 13 |
|
| 14 |
def resize_with_padding(img, expected_size):
|
| 15 |
img.thumbnail((expected_size[0], expected_size[1]))
|
|
|
|
| 9 |
# Initialize the Diffusion Pipeline
|
| 10 |
model_id = "yahoo-inc/photo-background-generation"
|
| 11 |
pipeline = DiffusionPipeline.from_pretrained(model_id, custom_pipeline=model_id)
|
| 12 |
+
pipeline = pipeline.to('cpu')
|
| 13 |
|
| 14 |
def resize_with_padding(img, expected_size):
|
| 15 |
img.thumbnail((expected_size[0], expected_size[1]))
|