Spaces:
Sleeping
Sleeping
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):
|