Update app.py

app.py

@@ -9,8 +9,10 @@ import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
+# Description de la démo
 DESCRIPTION = "# Mistral-7B v0.3"
 
+# Si pas de GPU détecté, afficher un message
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
@@ -20,20 +22,27 @@ MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
 
 if torch.cuda.is_available():
     model_id = "mistralai/Mistral-7B-Instruct-v0.3"
-    model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
+    model = AutoModelForCausalLM.from_pretrained(
+        model_id,
+        torch_dtype=torch.float16,
+        device_map="auto"
+    )
     tokenizer = AutoTokenizer.from_pretrained(model_id)
 
-
 @spaces.GPU
 def generate(
     message: str,
     chat_history: list[dict],
+    # Valeurs par défaut pour la génération
     max_new_tokens: int = 1024,
     temperature: float = 0.6,
     top_p: float = 0.9,
     top_k: int = 50,
     repetition_penalty: float = 1.2,
 ) -> Iterator[str]:
+    """
+    Génération de texte à partir de l'historique de conversation (chat_history) et du message utilisateur.
+    """
     conversation = [*chat_history, {"role": "user", "content": message}]
 
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
@@ -42,7 +51,12 @@ def generate(
         gr.Warning(f"Trimmed input from conversation as it was longer than {MAX_INPUT_TOKEN_LENGTH} tokens.")
     input_ids = input_ids.to(model.device)
 
-    streamer = TextIteratorStreamer(tokenizer, timeout=20.0, skip_prompt=True, skip_special_tokens=True)
+    streamer = TextIteratorStreamer(
+        tokenizer,
+        timeout=20.0,
+        skip_prompt=True,
+        skip_special_tokens=True
+    )
     generate_kwargs = dict(
         {"input_ids": input_ids},
         streamer=streamer,
@@ -62,57 +76,33 @@ def generate(
         outputs.append(text)
         yield "".join(outputs)
 
+# CSS pastel (dégradé sur toute la page)
+custom_css = """
+.gradio-container {
+    background: linear-gradient(135deg, #FDE2E2, #E2ECFD);
+}
+"""
+
+# Liste des questions prédéfinies
+predefined_examples = [
+    ["I - C’est quoi le consentement ? Comment savoir si ma copine a envie de moi ?"],
+    ["II - C’est quoi une agression sexuelle ?"],
+    ["III - C’est quoi un viol ?"],
+    ["IV - C’est quoi un attouchement ?"],
+    ["V - C’est quoi un harcèlement sexuel ?"],
+    ["VI - Est ce illégal de visionner du porno ?"],
+    ["VII - Mon copain me demande un nude, dois-je le faire ?"],
+    ["VIII - Mon ancien copain me menace de poster des photos de moi nue sur internet, que faire ?"],
+    ["IX - Que puis-je faire si un membre de ma famille me touche d’une manière bizarre, mais que j’ai peur de parler ou de ne pas être cru ?"],
+]
 
+# Création de l'interface de chat
 demo = gr.ChatInterface(
     fn=generate,
-    additional_inputs=[
-        gr.Slider(
-            label="Max new tokens",
-            minimum=1,
-            maximum=MAX_MAX_NEW_TOKENS,
-            step=1,
-            value=DEFAULT_MAX_NEW_TOKENS,
-        ),
-        gr.Slider(
-            label="Temperature",
-            minimum=0.1,
-            maximum=4.0,
-            step=0.1,
-            value=0.6,
-        ),
-        gr.Slider(
-            label="Top-p (nucleus sampling)",
-            minimum=0.05,
-            maximum=1.0,
-            step=0.05,
-            value=0.9,
-        ),
-        gr.Slider(
-            label="Top-k",
-            minimum=1,
-            maximum=1000,
-            step=1,
-            value=50,
-        ),
-        gr.Slider(
-            label="Repetition penalty",
-            minimum=1.0,
-            maximum=2.0,
-            step=0.05,
-            value=1.2,
-        ),
-    ],
-    stop_btn=None,
-    examples=[
-        ["Hello there! How are you doing?"],
-        ["Can you explain briefly to me what is the Python programming language?"],
-        ["Explain the plot of Cinderella in a sentence."],
-        ["How many hours does it take a man to eat a Helicopter?"],
-        ["Write a 100-word article on 'Benefits of Open-Source in AI research'"],
-    ],
     type="messages",
     description=DESCRIPTION,
-    css_paths="style.css",
+    css=custom_css,
+    examples=predefined_examples,
 )
 
 if __name__ == "__main__":