Refactoring, ajout des logos drane et lucie

.gitignore

@@ -1 +1,2 @@
 git-commit-push.bat
+

.python-version

@@ -0,0 +1 @@
+3.11

README.md

@@ -9,6 +9,7 @@ app_file: app.py
 pinned: false
 license: mit
 short_description: Chat with Lucie
+python_version: 3.11
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,3 +1,7 @@
+'''
+    Inspiré de https://huggingface.co/spaces/Tonic/Lucie-7B
+'''
+
 import gradio as gr
 import spaces
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -37,14 +41,15 @@ with gr.Blocks(title="Lucie",
         Tu es honnête et si tu ne sais pas quelque chose, tu le dis simplement.'''
         full_prompt = f"""<|system|>{system_prompt}</s><|user|>{question}</s><|assistant|>"""
         inputs = TOKENIZER(full_prompt, return_tensors="pt").to(DEVICE)
+        # Tous les paramètres sont les paramètres par défaut de Tonic/Lucie-7B
         outputs = MODEL.generate(
             **inputs,
             # max_new_tokens=max_new_tokens, # TODO: S'occuper des max_tokens avec tous les modèles
             max_new_tokens=512,
-            # temperature=temperature,
-            # top_p=top_p,
-            # top_k=top_k,
-            # repetition_penalty=repetition_penalty,
+            temperature=0.7,
+            top_p=0.9,
+            top_k=50,
+            repetition_penalty=1.2,
             do_sample=True,
             pad_token_id=TOKENIZER.eos_token_id
         )
@@ -52,29 +57,44 @@ with gr.Blocks(title="Lucie",
         r = response.split("<|assistant|>")[-1].strip()
         return r, gr.Image()
 
-    with gr.Tab("Chat"):
-        with gr.Row():
-            gr.Markdown('''# Lucie d'OpenLLM
-                        
-                        ## Discute avec Lucie
-                        ''')
-        with gr.Row(): 
-            gr.HTML('''<div><p align='right'>Pose ta question:</p></div>''')
-            question = gr.Textbox("", show_copy_button=False,
-                       show_label=False,
-                       container=False)
-            send_btn = gr.Button("Ok")
+    with gr.Row():
+        gr.Image("./files/drane.png",  show_download_button=False, 
+                show_fullscreen_button=False, show_label=False, show_share_button=False,
+                interactive=False, container=False)
+        # https://www.svgrepo.com/svg/403600/girl
+        gr.Image("./files/lucie.png",  show_download_button=False, 
+                show_fullscreen_button=False, show_label=False, show_share_button=False,
+                interactive=False, container=False)
+
+    with gr.Row():
+        gr.Markdown("# Lucie d'OpenLLM")
+        gr.Markdown("## Discute avec Lucie")
+
+        # gr.HTML('''<div><p align='right'>Pose ta question:</p></div>''')
+    with gr.Row():
+        question = gr.Textbox(
+            "", 
+            placeholder="Pose ta question ici",
+            show_copy_button=False,
+            show_label=False,
+            container=False,
+            lines=3,
+            autofocus=True,
+            scale=10
+        )
+        send_btn = gr.Button("Ok", scale=1)
         # L'image blanche qui affiche la progression 
-        wait = gr.Image("./files/white.jpg", height=25, show_download_button=False, 
-                    show_fullscreen_button=False, show_label=False, show_share_button=False,
-                    interactive=False, container=False, visible=True) 
+        # wait = gr.Image("./files/white.jpg", height=25, show_download_button=False, 
+        #             show_fullscreen_button=False, show_label=False, show_share_button=False,
+        #             interactive=False, container=False, visible=True) 
         
         resp = gr.Textbox("", show_copy_button=False,
                        show_label=False,
                        container=False,
                        max_lines=15)
         
-    send_btn.click(send, inputs=[question], outputs=[resp, wait])
+    # send_btn.click(send, inputs=[question], outputs=[resp, wait])
+    send_btn.click(send, inputs=[question], outputs=[resp])
             
 if __name__ == "__main__":
     demo.queue().launch()