Update app.py

app.py

@@ -1,15 +1,13 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
 import re
 
 # ==================== CHARGEMENT DU MODÈLE ====================
 print("🔄 Chargement du modèle...")
 
+# VERSION SIMPLE SANS QUANTIZATION
 model = AutoModelForCausalLM.from_pretrained(
     "padufour/mistral-accor-emails",
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-    device_map="auto",
     low_cpu_mem_usage=True
 )
 
@@ -21,7 +19,6 @@ print("✅ Modèle chargé et prêt !")
 def generate_email(email_type, client_name, company, objective, 
                    segment, priority, history, style, language):
     
-    # Construire le prompt
     prompt = f"""### Instruction : Rédige un email du type "{email_type}" pour un client professionnel. L'email doit être complet, sans placeholder ni crochet [], et ne doit PAS mentionner le segment de clientèle dans le contenu.
 
 ### Input :
@@ -38,25 +35,22 @@ Important: Ne pas inclure de placeholders [], ne pas mentionner le segment, ne p
 
 ### Réponse :"""
     
-    # Générer
-    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+    inputs = tokenizer(prompt, return_tensors="pt")
     
-    with torch.no_grad():
-        outputs = model.generate(
-            **inputs,
-            max_new_tokens=800,
-            temperature=0.7,
-            top_p=0.9,
-            top_k=50,
-            repetition_penalty=1.1,
-            do_sample=True,
-            pad_token_id=tokenizer.pad_token_id,
-            eos_token_id=tokenizer.eos_token_id
-        )
+    outputs = model.generate(
+        **inputs,
+        max_new_tokens=800,
+        temperature=0.7,
+        top_p=0.9,
+        top_k=50,
+        repetition_penalty=1.1,
+        do_sample=True,
+        pad_token_id=tokenizer.pad_token_id,
+        eos_token_id=tokenizer.eos_token_id
+    )
     
     email = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
-    # Extraire la réponse
     if "### Réponse :" in email:
         email = email.split("### Réponse :")[1].strip()
     
@@ -66,11 +60,9 @@ Important: Ne pas inclure de placeholders [], ne pas mentionner le segment, ne p
     email = re.sub(r'at\s+[\w\.-]+@[\w\.-]+\.\w+', '', email)
     email = re.sub(r'or\s+\+?\d[\d\s\(\)\-\.]+', '', email)
     
-    # Ajouter signature
     if "ACCOR Team" not in email and "Accor Team" not in email:
         email += "\n\nBest regards,\nThe ACCOR Team"
     
-    # Nettoyer espaces
     email = re.sub(r'\n\s*\n\s*\n', '\n\n', email)
     email = re.sub(r'  +', ' ', email)
     
@@ -95,34 +87,29 @@ with gr.Blocks(theme=gr.themes.Soft(), title="ACCOR Email Generator") as demo:
             email_type = gr.Dropdown(
                 choices=["Newsletter", "New offers", "theme", "solution", "Hotel opening"],
                 value="Newsletter",
-                label="Email Type *",
-                info="Type of marketing campaign"
+                label="Email Type *"
             )
             
             segment = gr.Dropdown(
                 choices=["MICE", "AHBO", "leisure group"],
                 value="MICE",
-                label="Segment",
-                info="Customer segment (used internally, not displayed)"
+                label="Segment"
             )
             
             gr.Markdown("### 👤 Client Information")
             
             client_name = gr.Textbox(
                 label="Client Name *",
-                placeholder="Marie Dupont",
                 value="Marie Dupont"
             )
             
             company = gr.Textbox(
                 label="Company *",
-                placeholder="Event Solutions SA",
                 value="Event Solutions SA"
             )
             
             objective = gr.Textbox(
                 label="Objective *",
-                placeholder="Promote winter events",
                 value="Promote winter events",
                 lines=2
             )
@@ -137,13 +124,11 @@ with gr.Blocks(theme=gr.themes.Soft(), title="ACCOR Email Generator") as demo:
             
             history = gr.Textbox(
                 label="Client History",
-                placeholder="New customer",
                 value="New customer"
             )
             
             style = gr.Textbox(
                 label="Style",
-                placeholder="Professional and engaging",
                 value="Professional and engaging"
             )
             
@@ -164,28 +149,11 @@ with gr.Blocks(theme=gr.themes.Soft(), title="ACCOR Email Generator") as demo:
                 lines=25,
                 show_copy_button=True
             )
-            
-            gr.Markdown(
-                """
-                ---
-                **Model:** [padufour/mistral-accor-emails](https://huggingface.co/padufour/mistral-accor-emails)
-                
-                **Tips for best results:**
-                - Provide detailed objectives
-                - Specify client history for personalization
-                - Use descriptive style preferences
-                """
-            )
-    
-    # Exemples
-    gr.Markdown("### 💡 Quick Examples")
     
     gr.Examples(
         examples=[
             ["Newsletter", "Marie Dupont", "Event Solutions SA", "Promote winter events", "MICE", "High", "Long-term client", "Professional and warm", "ENGLISH"],
-            ["New offers", "Pierre Martin", "Travel Agency Plus", "Showcase summer destinations", "leisure group", "Medium", "Regular customer", "Friendly and inspiring", "ENGLISH"],
-            ["theme", "Sophie Laurent", "Corporate Events Ltd", "Present emerging trends in 2025", "MICE", "High", "Industry leader", "Modern and innovative", "ENGLISH"],
-            ["solution", "Thomas Bernard", "Business Travel Corp", "Promote loyalty program benefits", "AHBO", "High", "Frequent traveler", "Professional and convincing", "ENGLISH"],
+            ["New offers", "Pierre Martin", "Travel Agency Plus", "Showcase summer destinations", "leisure group", "Medium", "Regular customer", "Friendly", "ENGLISH"],
         ],
         inputs=[email_type, client_name, company, objective, segment, priority, history, style, language],
         outputs=output,
@@ -193,12 +161,10 @@ with gr.Blocks(theme=gr.themes.Soft(), title="ACCOR Email Generator") as demo:
         cache_examples=False,
     )
     
-    # Event
     generate_btn.click(
         fn=generate_email,
         inputs=[email_type, client_name, company, objective, segment, priority, history, style, language],
         outputs=output
     )
 
-# Lancer
 demo.launch()