Update utils.py

utils.py

@@ -14,7 +14,7 @@ def generate_story(prompt, duration):
     text = tokenizer(
         f"Write a detailed story of {duration} minutes about: {prompt}", 
         return_tensors="pt"
-    ).to("cuda")
+    ).to("cpu")
 
     output = model.generate(**text, max_new_tokens=900)
     story = tokenizer.decode(output[0], skip_special_tokens=True)
@@ -37,7 +37,7 @@ def generate_images(story):
     sd = StableDiffusionPipeline.from_pretrained(
         "runwayml/stable-diffusion-v1-5",
         torch_dtype=torch.float16
-    ).to("cuda")
+    ).to("cpu")
 
     scenes = story.split(".")[:10]   # create 10 images max
     paths = []