Update space

model/waifu.py

@@ -23,17 +23,14 @@
 from diffusers import StableDiffusionPipeline
 import torch
 
-
-def generate_waifu(prompt: str, model_id):
-    model_id = "hakurei/waifu-diffusion"
-    pipe = StableDiffusionPipeline.from_pretrained(
-        model_id,
-        torch_dtype=torch.float32,         # CPU는 float32만 지원
-        use_safetensors=False
-        # revision=None                       # revision="fp16" 삭제
-    ).to("cpu")                            # 반드시 CPU로 지정
-
-
+model_id = "hakurei/waifu-diffusion"
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id,
+    torch_dtype=torch.float32,
+    use_safetensors=False
+).to("cpu")  # CPU로 명시
+
+def generate_waifu(prompt: str):
     image = pipe(prompt=prompt).images[0]
     return image