Test code online commit

app.py

@@ -22,13 +22,10 @@ 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,
@@ -66,6 +63,7 @@ def generate(
         outputs.append(text)
         yield "".join(outputs)
 
+
 # CSS pour appliquer le dégradé pastel à TOUTE la page
 custom_css = """
 html, body {
@@ -78,7 +76,7 @@ html, body {
 
 # Questions prédéfinies
 predefined_examples = [
-    ["1 - C’est quoi le consentement ? Comment savoir si ma copine a envie de moi ?"],
+    ["1 - C’est quoi le consentement ? Comment savoir si ma copine a envie de moi ?"],  # noqa: RUF001
     ["2 - C’est quoi une agression sexuelle ?"],
     ["3 - C’est quoi un viol ?"],
     ["4 - C’est quoi un attouchement ?"],
@@ -86,15 +84,17 @@ predefined_examples = [
     ["6 - Est ce illégal de visionner du porno ?"],
     ["7 - Mon copain me demande un nude, dois-je le faire ?"],
     ["8 - Mon ancien copain me menace de poster des photos de moi nue sur internet, que faire ?"],
-    ["9 - 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 ?"],
+    [
+        "9 - 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 ?"
+    ],
 ]
 
 demo = gr.ChatInterface(
     fn=generate,
     type="messages",
     description=DESCRIPTION,
-    css=custom_css,       # On applique le CSS pastel global
-    examples=predefined_examples
+    css=custom_css,  # On applique le CSS pastel global
+    examples=predefined_examples,
 )
 
 if __name__ == "__main__":