Update app.py

app.py

@@ -5,8 +5,8 @@ import requests
 from io import BytesIO
 import os
 
-# Token Hugging Face
-HF_TOKEN = "hf_..." # Assurez-vous que ce token est correct
+# Token Hugging Face - Assurez-vous que ce token est correct et a les bonnes permissions
+HF_TOKEN = "hf_..."  # Remplacez par votre vrai token
 
 # Styles disponibles
 STYLES = {
@@ -62,8 +62,8 @@ def generate_image(prompt, style, format_size, material):
         "negative_prompt": "low quality, blurry, bad anatomy, distorted, disfigured, pixelated",
         "num_inference_steps": 40,
         "guidance_scale": 7.5,
-        "width": 768,
-        "height": 768
+        "width": 512,
+        "height": 512
     }
 
     try:
@@ -76,17 +76,20 @@ def generate_image(prompt, style, format_size, material):
             return Image.open(BytesIO(response.content))
         else:
             print(f"Erreur API - Response: {response.text}")
+            error_message = f"Erreur {response.status_code}: {response.text}"
             error_image = Image.new('RGB', (512, 512), color='red')
-            return error_image
+            return gr.Image.update(value=error_image, label=error_message)
             
     except requests.exceptions.RequestException as e:
         print(f"Erreur de requête : {str(e)}")
+        error_message = f"Erreur de requête : {str(e)}"
         error_image = Image.new('RGB', (512, 512), color='orange')
-        return error_image
+        return gr.Image.update(value=error_image, label=error_message)
     except Exception as e:
         print(f"Exception inattendue : {str(e)}")
+        error_message = f"Erreur inattendue : {str(e)}"
         error_image = Image.new('RGB', (512, 512), color='yellow')
-        return error_image
+        return gr.Image.update(value=error_image, label=error_message)
 
 def update_suggestions(style):
     """