Update app.py
Browse files
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 |
-
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
| 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
|