Spaces:
Runtime error
Runtime error
Commit
·
477dde5
1
Parent(s):
8fb9752
modelProblems
Browse files
app.py
CHANGED
|
@@ -20,9 +20,9 @@ import numpy as np
|
|
| 20 |
# ]
|
| 21 |
# current_model=model_box[0]
|
| 22 |
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
|
| 27 |
@spaces.GPU
|
| 28 |
def predict(steps, seed):
|
|
|
|
| 20 |
# ]
|
| 21 |
# current_model=model_box[0]
|
| 22 |
|
| 23 |
+
pipeline = DiffusionPipeline.from_pretrained("nathanReitinger/MNIST-diffusion-oneImage")
|
| 24 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 25 |
+
pipeline = pipeline.to(device=device)
|
| 26 |
|
| 27 |
@spaces.GPU
|
| 28 |
def predict(steps, seed):
|