Update app.py

app.py

@@ -3,21 +3,21 @@ import torch
 from diffusers import StableDiffusionImg2ImgPipeline
 from PIL import Image
 
-# مدل سبک، بدون توکن، سریع
-model_id = "nitrosocke/anything-v3-small"
+# مدل سبک و سریع
+model_id = "hakurei/waifu-diffusion-v1"
 
-# بارگذاری مدل روی GPU (اگر دارید) یا CPU
+# بارگذاری مدل روی GPU (اگر در دسترس باشد)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     model_id,
-    torch_dtype=torch.float16 if device=="cuda" else torch.float32
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32
 ).to(device)
 
 def img2anime(input_image, prompt, strength=0.7, steps=20, width=560, height=560):
     # تبدیل به RGB
     init_image = input_image.convert("RGB")
     # تغییر اندازه ورودی (حداکثر 850x850)
-    init_image = init_image.resize((min(width,850), min(height,850)))
+    init_image = init_image.resize((min(width, 850), min(height, 850)))
     
     result = pipe(
         prompt=prompt,
@@ -35,9 +35,9 @@ iface = gr.Interface(
         gr.Image(type="pil", label="ورودی عکس"),
         gr.Textbox(label="پرومپت انیمه"),
         gr.Slider(0.0, 1.0, value=0.7, label="Strength"),
-        gr.Slider(1,50, value=20, step=1, label="Steps"),
-        gr.Slider(128,850,value=560,step=1,label="Width"),
-        gr.Slider(128,850,value=560,step=1,label="Height")
+        gr.Slider(1, 50, value=20, step=1, label="Steps"),
+        gr.Slider(128, 850, value=560, step=1, label="Width"),
+        gr.Slider(128, 850, value=560, step=1, label="Height")
     ],
     outputs=gr.Image(type="pil", label="خروجی انیمه"),
     title="تبدیل عکس به انیمه سریع بدون توکن",