Bourgi82 commited on
Commit
36c42b2
·
1 Parent(s): 9a01d11
Files changed (2) hide show
  1. app.py +53 -0
  2. requierements.txt +5 -0
app.py ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ from datetime import datetime
3
+
4
+ # Charger le pipeline de génération de texte avec le modèle Llama
5
+ # Use a pipeline as a high-level helper
6
+ pipe = pipeline("text-generation", model="benjleite/t5-french-qa")
7
+ # Fonction pour obtenir une réponse avec le modèle
8
+ def get_llama_response(question: str) -> str:
9
+ current_date = datetime.now().strftime('%Y-%m-%d')
10
+
11
+ prompt = (
12
+ "Tu es un assistant francophone expert. "
13
+ f"\nNous sommes le {current_date}."
14
+ "\nRéponds avec concision et précision, en adoptant un ton professionnel. "
15
+ "Évite les phrases inutiles et n'invente rien si les données ne sont pas disponibles. "
16
+ f"\nQuestion: {question}"
17
+ )
18
+
19
+ # Utilisation du pipeline pour générer une réponse
20
+ response = pipe(prompt, max_length=200, num_return_sequences=1)
21
+
22
+ # La réponse générée est dans le champ 'generated_text' de la sortie
23
+ return response[0]['generated_text']
24
+
25
+ # Fonction qui orchestre la réponse du chatbot en utilisant la question
26
+ def chatbot_respond(message):
27
+ response = get_llama_response(message)
28
+ return "", [(message, response)]
29
+
30
+ # Création de l'interface Gradio
31
+ with gr.Blocks() as demo:
32
+ gr.Markdown("## Chatbot d'analyse de performance réseau")
33
+
34
+ # Composant affichant l'historique de la conversation
35
+ chatbot_display = gr.Chatbot(label="Conversation")
36
+
37
+ # Champ de saisie pour la question utilisateur
38
+ message_input = gr.Textbox(label="Votre question", placeholder="Tapez votre question ici...", lines=1)
39
+
40
+ # Bouton pour envoyer la question
41
+ send_btn = gr.Button("Envoyer")
42
+
43
+ # Liaison du bouton à la fonction chatbot_respond
44
+ send_btn.click(
45
+ fn=chatbot_respond,
46
+ inputs=message_input,
47
+ outputs=[message_input, chatbot_display],
48
+ api_name="chatbot_respond",
49
+ queue=True
50
+ )
51
+
52
+ # Lancement de l'interface
53
+ demo.launch()
requierements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ gradio
2
+ transformers
3
+ torch
4
+ numpy
5
+ pandas