Update space

model/animagine_xl.py

@@ -10,12 +10,14 @@ def generate_animagine_xl(prompt: str):
         variant="fp16"
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    pipe = pipe.to("cuda")
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    pipe = pipe.to(device)
     image = pipe(prompt=prompt, width=1024, height=1024).images[0]
     image.save("output_animagine_xl.png")
     print("✅ 저장 완료: output_animagine_xl.png")
     return image
 
+
 if __name__ == "__main__":
     prompt = "그녀를 바라보는 한 남자의 야망"
     generate_animagine_xl(prompt)