Update app.py

app.py

@@ -1,14 +1,15 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 import re
 
 print("🔄 Chargement du modèle...")
 
-# CHARGER SANS QUANTIZATION
+# IGNORER LA QUANTIZATION avec un objet vide
 model = AutoModelForCausalLM.from_pretrained(
     "padufour/mistral-accor-emails",
     low_cpu_mem_usage=True,
-    quantization_config=None  # ← Ignore la config 4-bit du modèle
+    device_map=None,  # Pas de device_map auto
+    trust_remote_code=True
 )
 
 tokenizer = AutoTokenizer.from_pretrained("padufour/mistral-accor-emails")
@@ -63,51 +64,32 @@ Important: Ne pas inclure de placeholders [], ne pas mentionner le segment, ne p
     email = re.sub(r'\n\s*\n\s*\n', '\n\n', email)
     return email.strip()
 
-with gr.Blocks(theme=gr.themes.Soft(), title="ACCOR Email Generator") as demo:
-    
-    gr.Markdown("# 🎯 ACCOR Email Generator\n**Powered by Mistral-7B Fine-tuned**")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎯 ACCOR Email Generator\n**Powered by Mistral-7B**")
     
     with gr.Row():
         with gr.Column():
             email_type = gr.Dropdown(
-                choices=["Newsletter", "New offers", "theme", "solution"],
-                value="Newsletter",
-                label="Email Type"
-            )
-            segment = gr.Dropdown(
-                choices=["MICE", "AHBO", "leisure group"],
-                value="MICE",
-                label="Segment"
+                ["Newsletter", "New offers", "theme", "solution"],
+                value="Newsletter", label="Email Type"
             )
             client_name = gr.Textbox(label="Client Name", value="Marie Dupont")
-            company = gr.Textbox(label="Company", value="Event Solutions SA")
-            objective = gr.Textbox(label="Objective", value="Promote winter events", lines=2)
+            company = gr.Textbox(label="Company", value="Event Solutions")
+            objective = gr.Textbox(label="Objective", value="Promote events", lines=2)
             
-            with gr.Accordion("Advanced Options", open=False):
-                priority = gr.Radio(["High", "Medium", "Low"], value="High", label="Priority")
-                history = gr.Textbox(label="Client History", value="New customer")
-                style = gr.Textbox(label="Style", value="Professional")
-                language = gr.Radio(["ENGLISH", "FRENCH"], value="ENGLISH", label="Language")
+            segment = gr.Dropdown(["MICE", "AHBO", "leisure group"], value="MICE", label="Segment")
+            priority = gr.Radio(["High", "Medium", "Low"], value="High", label="Priority")
+            history = gr.Textbox(label="History", value="New customer")
+            style = gr.Textbox(label="Style", value="Professional")
+            language = gr.Radio(["ENGLISH", "FRENCH"], value="ENGLISH")
             
-            generate_btn = gr.Button("🚀 Generate Email", variant="primary")
+            btn = gr.Button("🚀 Generate", variant="primary")
         
         with gr.Column():
-            output = gr.Textbox(label="Generated Email", lines=25, show_copy_button=True)
+            output = gr.Textbox(label="Email", lines=30, show_copy_button=True)
     
-    generate_btn.click(
-        fn=generate_email,
-        inputs=[email_type, client_name, company, objective, segment, priority, history, style, language],
-        outputs=output
-    )
-    
-    gr.Examples(
-        examples=[
-            ["Newsletter", "Marie Dupont", "Event Solutions", "Promote events", "MICE", "High", "New", "Professional", "ENGLISH"],
-            ["New offers", "John Smith", "Travel Corp", "Summer deals", "leisure group", "Medium", "Regular", "Friendly", "ENGLISH"],
-        ],
-        inputs=[email_type, client_name, company, objective, segment, priority, history, style, language],
-        outputs=output,
-        fn=generate_email,
-    )
+    btn.click(generate_email, 
+              [email_type, client_name, company, objective, segment, priority, history, style, language],
+              output)
 
 demo.launch()