kevtico20 commited on
Commit
072a251
Β·
1 Parent(s): c5635b2

fix: pasar HF_TOKEN como auth header al llamar OOTDiffusion (cuota ZeroGPU)

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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),