Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 import numpy as np
 import torch
 import safetensors.torch as sf
-
 from PIL import Image
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 from diffusers import AutoencoderKL, UNet2DConditionModel, DDIMScheduler, EulerAncestralDiscreteScheduler, DPMSolverMultistepScheduler
@@ -12,13 +11,7 @@ from diffusers.models.attention_processor import AttnProcessor2_0
 from transformers import CLIPTextModel, CLIPTokenizer
 from briarmbg import BriaRMBG
 from torch.hub import download_url_to_file
-from spaces import ZeroGPU
-
-# مقداردهی اولیه ZeroGPU
-ZeroGPU.initialize()
-
-# نصب کتابخانه‌های مورد نیاز در صورت اجرا روی Hugging Face Spaces
-os.system("pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121")
+from huggingface_hub import hf_hub_download
 
 # بررسی دسترسی به GPU
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -31,9 +24,12 @@ else:
 # دانلود و بارگذاری مدل
 model_path = './models/iclight_sd15_fc.safetensors'
 if not os.path.exists(model_path):
-    download_url_to_file(
-        url='https://huggingface.co/lllyasviel/ic-light/resolve/main/iclight_sd15_fc.safetensors',
-        dst=model_path
+    os.makedirs(os.path.dirname(model_path), exist_ok=True)
+    hf_hub_download(
+        repo_id='lllyasviel/ic-light',
+        filename='iclight_sd15_fc.safetensors',
+        local_dir='./models',
+        local_dir_use_symlinks=False
     )
 
 # بارگذاری مدل‌های مورد نیاز
@@ -119,4 +115,4 @@ with block:
     
     submit.click(fn=process_image, inputs=[prompt, width, height, steps, guidance], outputs=result_image)
 
-block.launch(server_name='0.0.0.0')
+block.launch(server_name='0.0.0.0', server_port=7860)