andrewchwalik
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ print(f"Using Hugging Face Token: {os.getenv('HUGGINGFACE_TOKEN')}")
|
|
11 |
model = StableDiffusionPipeline.from_pretrained(
|
12 |
"runwayml/stable-diffusion-v1-5"
|
13 |
)
|
14 |
-
model.to("
|
15 |
|
16 |
def generate_video(prompt):
|
17 |
# Generate frames
|
|
|
11 |
model = StableDiffusionPipeline.from_pretrained(
|
12 |
"runwayml/stable-diffusion-v1-5"
|
13 |
)
|
14 |
+
model.to("cpu") # Use CPU since GPU is not available
|
15 |
|
16 |
def generate_video(prompt):
|
17 |
# Generate frames
|