Hugging Face's logo

Spaces:
zahidpichen
/
Story_Board
Sleeping

App Files Community
zahidpichen commited on
Commit
989b426
·
verified ·
1 Parent(s): 5f65efb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def split_story(story):
42
 
43
  def generate_images(story_parts, save_dir="story_images"):
44
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
45
- pipe = pipe.to("cuda")
46
  os.makedirs(save_dir, exist_ok=True)
47
  image_paths = []
48
  for i, part in enumerate(story_parts):
 
42
 
43
  def generate_images(story_parts, save_dir="story_images"):
44
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo")
45
+ pipe = pipe.to("cpu")
46
  os.makedirs(save_dir, exist_ok=True)
47
  image_paths = []
48
  for i, part in enumerate(story_parts):