Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|