Update app.py

app.py

@@ -1,29 +1,25 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionXLPipeline
+from diffusers import StableDiffusionPipeline
 
-# Load SDXL model
-model_id = "stabilityai/stable-diffusion-xl-base-1.0"
-pipe = StableDiffusionXLPipeline.from_pretrained(
+# 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,
-    safety_checker=None  # Disable safety checker for hackathon demo
+    torch_dtype=torch.float16
 )
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Function to generate outfit image + suggestion
+# Fonction pour générer l'image + suggestion
 def generate_outfit(weather, activity, style):
     try:
-        # Build prompt
         prompt = (
             f"A realistic full-body outfit for {weather} weather, "
-            f"{activity} activity, {style} style, front view, 30-degree angle, highly detailed, full body"
+            f"{activity} activity, {style} style, front view, full body, fashion style"
         )
-        # SDXL generates output as a list of PIL images
         result = pipe(prompt, guidance_scale=7.5)
-        image = result.images[0]  # take the first image
+        image = result.images[0]  # Toujours prendre la première image
 
-        # Outfit suggestion
         suggestion_text = (
             f"💡 Suggested outfit for {weather} weather, {activity} activity, {style} style:\n"
             "- Top: light breathable shirt or jacket depending on weather\n"
@@ -32,11 +28,10 @@ def generate_outfit(weather, activity, style):
             "- Accessories: hat, sunglasses, or scarf depending on weather"
         )
         return image, suggestion_text
-
     except Exception as e:
         return None, f"❌ Error generating outfit: {str(e)}"
 
-# Gradio UI
+# UI Gradio
 custom_css = """
 body {background: linear-gradient(135deg, #232526, #414345); color: white; font-family: 'Poppins', sans-serif;}
 .gradio-container {max-width: 900px !important; margin: auto;}
@@ -47,8 +42,8 @@ h1,h2,h3 {text-align:center; color:#fff;}
 """
 
 with gr.Blocks(css=custom_css) as app:
-    gr.Markdown("<h1 style='text-align:center;'>👗 ClothCast - AI Outfit Forecaster</h1>")
-    gr.Markdown("<p style='text-align:center;'>Select your weather, activity, and style to generate an AI outfit!</p>")
+    gr.Markdown("<h1 style='text-align:center;'>👗 ClothCast - AI Outfit Forecaster (Fast)</h1>")
+    gr.Markdown("<p style='text-align:center;'>Select your weather, activity, and style to generate an AI outfit quickly!</p>")
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -78,3 +73,4 @@ with gr.Blocks(css=custom_css) as app:
 
 app.launch(share=True)
 
+