Simonc-44 commited on
Commit
7a6bdbf
·
verified ·
1 Parent(s): 3ad6837

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -11,7 +11,8 @@ app = FastAPI()
11
 
12
  # Token HF récupéré depuis les variables d'environnement (Secrets)
13
  HF_TOKEN = os.environ.get("HF_TOKEN")
14
- API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 
15
 
16
  class ImageRequest(BaseModel):
17
  prompt: str
@@ -30,7 +31,9 @@ async def generate_image(request: ImageRequest):
30
  response = requests.post(API_URL, headers=headers, json=payload)
31
 
32
  if response.status_code != 200:
33
- raise Exception(f"Erreur API HF: {response.text}")
 
 
34
 
35
  # L'API retourne l'image binaire
36
  image_bytes = response.content
 
11
 
12
  # Token HF récupéré depuis les variables d'environnement (Secrets)
13
  HF_TOKEN = os.environ.get("HF_TOKEN")
14
+ # NOUVELLE URL (router.huggingface.co)
15
+ API_URL = "https://router.huggingface.co/hf-inference/models/stabilityai/stable-diffusion-xl-base-1.0"
16
 
17
  class ImageRequest(BaseModel):
18
  prompt: str
 
31
  response = requests.post(API_URL, headers=headers, json=payload)
32
 
33
  if response.status_code != 200:
34
+ # On logue le contenu complet pour debug
35
+ print(f"Erreur API: {response.text}")
36
+ raise Exception(f"Erreur API HF ({response.status_code}): {response.text}")
37
 
38
  # L'API retourne l'image binaire
39
  image_bytes = response.content