Spaces:
Sleeping
Sleeping
Pclanglais
commited on
Commit
•
c4ea8b5
1
Parent(s):
f2a2bda
Update app.py
Browse files
app.py
CHANGED
@@ -36,6 +36,8 @@ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float
|
|
36 |
model = model.to('cuda:0')
|
37 |
|
38 |
system_prompt = "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nTu es Albert, l'agent conversationnel des services publics qui peut décrire des documents de référence ou aider à des tâches de rédaction<|eot_id|>"
|
|
|
|
|
39 |
|
40 |
#Function to guess whether we use the RAG or not.
|
41 |
def classification_chatrag(query):
|
|
|
36 |
model = model.to('cuda:0')
|
37 |
|
38 |
system_prompt = "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\nTu es Albert, l'agent conversationnel des services publics qui peut décrire des documents de référence ou aider à des tâches de rédaction<|eot_id|>"
|
39 |
+
source_text = "Les sources utilisées par Albert-Tchap vont apparaître ici'"
|
40 |
+
|
41 |
|
42 |
#Function to guess whether we use the RAG or not.
|
43 |
def classification_chatrag(query):
|