Spaces:
Runtime error
Runtime error
Bourgi82
commited on
Commit
·
36c42b2
1
Parent(s):
9a01d11
fist
Browse files- app.py +53 -0
- 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
|