andrewchwalik commited on
Commit
1004b05
·
verified ·
1 Parent(s): d4f70fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("cuda") # Use GPU for faster processing (if available)
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