Hugging Face's logo

Spaces:
zahidpichen
/
Story_Board
Sleeping

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -42,7 +42,6 @@ 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("cpu")
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
  os.makedirs(save_dir, exist_ok=True)
46
  image_paths = []
47
  for i, part in enumerate(story_parts):