Update app.py

app.py

@@ -16,6 +16,7 @@ import chromadb
 from unidecode import unidecode
 
 from transformers import AutoTokenizer
+from transformers import pipeline
 import transformers
 import torch
 import tqdm
@@ -96,6 +97,9 @@ def initialize_llmchain(temperature, max_tokens, top_k, vector_db):
 
 qa = initialize_llmchain(0.6, 1024, 40, vec_cre) #The model question answer
 
+pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr") # This pipeline translate english to french , it isn't adviced as it add more latency
+
+
 def format_chat_history(message, chat_history):
     formatted_chat_history = []
     for user_message, bot_message in chat_history:
@@ -119,8 +123,13 @@ def conversation(message, history):
     response_source1_page = response_sources[0].metadata["page"] + 1
     response_source2_page = response_sources[1].metadata["page"] + 1
     response_source3_page = response_sources[2].metadata["page"] + 1
+    #If you want the return in english leave it at :
+    #return response_answer
+
+    #If you want the return in french
+    return pipe(response_answer)[0]['translation_text'] + " (Traduis d'anglais en français)"
+    
 
-    return response_answer