fix: force png output

app.py

@@ -20,25 +20,35 @@ def generate_image(prompt):
     Génère une image à partir du prompt en utilisant le modèle Hugging Face.
     """
     if pipe is None:
-        return "Le modèle n'a pas pu être chargé."
+        raise ValueError("The model couldn't be loaded.")
 
     try:
         # Générer l'image
-        image = pipe(prompt).images[0]
+        result = pipe(prompt)
+        
+        # Vérifier que le résultat contient des images
+        if not hasattr(result, 'images') or len(result.images) == 0:
+            raise ValueError("The model couldn't generate an image.")
+
+        image = result.images[0]
+
+        # S'assurer que l'image est au format PIL.Image
+        if not isinstance(image, Image.Image):
+            image = Image.fromarray(image)
 
         return image
 
     except Exception as e:
-        # Gestion des erreurs avec un retour informatif
-        return f"Erreur lors de la génération : {str(e)}"
+        # Lever une exception pour que Gradio puisse la gérer
+        raise ValueError(f"Erreur lors de la génération : {str(e)}")
 
 # Interface Gradio
 iface = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Prompt"),
-    outputs=gr.Image(label="Image générée"),
-    title="Générateur d'images RickTheNPC",
-    description="Entrez un prompt et générez une image avec le modèle RickTheNPC."
+    outputs=gr.Image(label="Generated Image"),
+    title="Rick Generator",
+    description="Enter a prompt to generate an image with the Rick Generator model."
 )
 
 # Lancer l'application