mahdideveloepr commited on
Commit
ec03c85
·
verified ·
1 Parent(s): c67957d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -11,7 +11,9 @@ from diffusers.models.attention_processor import AttnProcessor2_0
11
  from transformers import CLIPTextModel, CLIPTokenizer
12
  from briarmbg import BriaRMBG
13
  from torch.hub import download_url_to_file
 
14
 
 
15
  # نصب کتابخانه‌های مورد نیاز در صورت اجرا روی Hugging Face Spaces
16
  os.system("pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121")
17
 
 
11
  from transformers import CLIPTextModel, CLIPTokenizer
12
  from briarmbg import BriaRMBG
13
  from torch.hub import download_url_to_file
14
+ from spaces import ZeroGPU
15
 
16
+ ZeroGPU.initialize()
17
  # نصب کتابخانه‌های مورد نیاز در صورت اجرا روی Hugging Face Spaces
18
  os.system("pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121")
19