Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ if os.path.exists(result_path):
|
|
26 |
|
27 |
|
28 |
device = "cpu"
|
29 |
-
model = Dalle.from_pretrained(
|
30 |
model.to(device=device)
|
31 |
|
32 |
model_clip, preprocess_clip = clip.load("ViT-B/32", device=device)
|
|
|
26 |
|
27 |
|
28 |
device = "cpu"
|
29 |
+
model = Dalle.from_pretrained(result_path) # This will automatically download the pretrained model.
|
30 |
model.to(device=device)
|
31 |
|
32 |
model_clip, preprocess_clip = clip.load("ViT-B/32", device=device)
|