Pclanglais commited on
Commit
c4ea8b5
1 Parent(s): f2a2bda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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):