Update app.py + style.css

Modification de la description, ajouts des 10 questions de base dans app.py
Dans le tyle.css, ajout d'un dégradé pastel

app.py

@@ -9,13 +9,11 @@ import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
-DESCRIPTION = "# Mistral-7B v0.3"
+DESCRIPTION = "# L'IA qui t'informe sur tes désirs"
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
-MAX_MAX_NEW_TOKENS = 2048
-DEFAULT_MAX_NEW_TOKENS = 1024
 MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
 
 if torch.cuda.is_available():
@@ -25,18 +23,22 @@ if torch.cuda.is_available():
 
 
 @spaces.GPU
-def generate(
-    message: str,
-    chat_history: list[dict],
-    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]:
-    conversation = [*chat_history, {"role": "user", "content": message}]
+def generate(message: str, chat_history: list[dict]) -> Iterator[str]:
+    """Fonction de génération sans sliders : les paramètres
+    de génération (max_new_tokens, température, etc.) sont
+    fixés en dur.
+    """
+    # Valeurs par défaut fixées
+    max_new_tokens = 1024
+    temperature = 0.6
+    top_p = 0.9
+    top_k = 50
+    repetition_penalty = 1.2
 
+    # Prépare la conversation
+    conversation = [*chat_history, {"role": "user", "content": message}]
     input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt")
+
     if input_ids.shape[1] > MAX_INPUT_TOKEN_LENGTH:
         input_ids = input_ids[:, -MAX_INPUT_TOKEN_LENGTH:]
         gr.Warning(f"Trimmed input from conversation as it was longer than {MAX_INPUT_TOKEN_LENGTH} tokens.")
@@ -63,52 +65,22 @@ def generate(
         yield "".join(outputs)
 
 
+# On ne fournit plus 'additional_inputs' ici, donc aucun slider ne sera affiché
 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'"],
+        ["C’est quoi le consentement ? Comment savoir si ma copine a envie de moi ?"],
+        ["C’est quoi une agression sexuelle ?"],
+        ["C’est quoi un viol ?"],
+        ["C’est quoi un attouchement ?"],
+        ["C’est quoi un harcèlement sexuel ?"],
+        ["Est-ce illégal de visionner du porno ?"],
+        ["C’est quoi un harcèlement sexuel ?"],
+        ["Mon copain me demande un nude, dois-je le faire ?"],
+        ["Mon ancien copain me menace de poster des photos de moi nue sur internet, que faire ?"],
+        ["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 ?"
+        ],
     ],
     type="messages",
     description=DESCRIPTION,