Spaces:
Paused
Paused
kevtico20 commited on
Commit Β·
072a251
1
Parent(s): c5635b2
fix: pasar HF_TOKEN como auth header al llamar OOTDiffusion (cuota ZeroGPU)
Browse files
app.py
CHANGED
|
@@ -7,6 +7,9 @@ from PIL import Image
|
|
| 7 |
from gradio_client import Client, handle_file
|
| 8 |
from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
|
| 9 |
|
|
|
|
|
|
|
|
|
|
| 10 |
# βββ Carga de modelos βββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 11 |
|
| 12 |
multiview_pipe = None
|
|
@@ -61,7 +64,7 @@ def tryon(person_image: Image.Image, garment_image: Image.Image) -> Image.Image:
|
|
| 61 |
garment_path = f_garment.name
|
| 62 |
|
| 63 |
try:
|
| 64 |
-
client = Client("eduardo4547/OOTDiffusion", verbose=False)
|
| 65 |
result = client.predict(
|
| 66 |
vton_img=handle_file(person_path),
|
| 67 |
garm_img=handle_file(garment_path),
|
|
|
|
| 7 |
from gradio_client import Client, handle_file
|
| 8 |
from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
|
| 9 |
|
| 10 |
+
HF_TOKEN = os.environ.get("HF_TOKEN", "")
|
| 11 |
+
_OOTD_HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else None
|
| 12 |
+
|
| 13 |
# βββ Carga de modelos βββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 14 |
|
| 15 |
multiview_pipe = None
|
|
|
|
| 64 |
garment_path = f_garment.name
|
| 65 |
|
| 66 |
try:
|
| 67 |
+
client = Client("eduardo4547/OOTDiffusion", headers=_OOTD_HEADERS, verbose=False)
|
| 68 |
result = client.predict(
|
| 69 |
vton_img=handle_file(person_path),
|
| 70 |
garm_img=handle_file(garment_path),
|