Update app.py

app.py

@@ -1,24 +1,25 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionPipeline
+from diffusers import DiffusionPipeline
 
-# RAPIDE: Utilisation d'un modèle plus léger pour la génération rapide
-model_id = "stabilityai/stable-diffusion-2-1-base"
-pipe = StableDiffusionPipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16
+# Load the new model
+# Use bfloat16 for speed and CUDA if available
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = DiffusionPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-dev", 
+    dtype=torch.bfloat16,
+    device_map=device
 )
-pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Fonction pour générer l'image + suggestion
+# Function to generate outfit image + suggestion
 def generate_outfit(weather, activity, style):
     try:
         prompt = (
             f"A realistic full-body outfit for {weather} weather, "
-            f"{activity} activity, {style} style, front view, full body, fashion style"
+            f"{activity} activity, {style} style, front view, fashion style, full body"
         )
-        result = pipe(prompt, guidance_scale=7.5)
-        image = result.images[0]  # Toujours prendre la première image
+        result = pipe(prompt)
+        image = result.images[0]
 
         suggestion_text = (
             f"💡 Suggested outfit for {weather} weather, {activity} activity, {style} style:\n"
@@ -31,7 +32,7 @@ def generate_outfit(weather, activity, style):
     except Exception as e:
         return None, f"❌ Error generating outfit: {str(e)}"
 
-# UI Gradio
+# Gradio UI
 custom_css = """
 body {background: linear-gradient(135deg, #232526, #414345); color: white; font-family: 'Poppins', sans-serif;}
 .gradio-container {max-width: 900px !important; margin: auto;}
@@ -74,3 +75,4 @@ with gr.Blocks(css=custom_css) as app:
 app.launch(share=True)
 
 
+