add files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.faiss filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,47 @@
+import gradio as gr
+import os
+import textwrap
+from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.llms import OpenAI
+from langchain.prompts import PromptTemplate
+from langchain.chains import RetrievalQA
+from langchain.llms import OpenAI
+from langchain.docstore.document import Document
+from langchain.vectorstores import FAISS
+
+# Function that sets up the generation model using the provided API key
+def setup_generator(api_key):
+    os.environ["OPENAI_API_KEY"] = api_key
+    embeddings = HuggingFaceEmbeddings(model_name="dangvantuan/sentence-camembert-base")
+    db = FAISS.load_local("faiss_index", embeddings)
+    
+    prompt_template = """Use the following pieces of context to answer the question at the end.
+    Give many informations as possible.
+    If you don't know the answer, just say that you don't know, don't try to make up an answer.
+
+    {context}
+
+    Question: {question}
+    Answer in French:"""
+    PROMPT = PromptTemplate(
+        template=prompt_template, input_variables=["context", "question"]
+    )
+    chain_type_kwargs = {"prompt": PROMPT}
+    qa = RetrievalQA.from_chain_type(llm=OpenAI(), chain_type="stuff",
+                                  retriever=db.as_retriever(), chain_type_kwargs=chain_type_kwargs)
+    return qa
+
+def generate_text(api_key, query):
+
+    
+    qa = setup_generator(api_key)
+    formatted_text = '\n'.join(textwrap.wrap(qa.run(query), width=60))
+    return formatted_text
+
+iface = gr.Interface(
+    fn=generate_text,
+    inputs=[gr.inputs.Textbox(lines=1, placeholder="Input API Key..."),
+            gr.inputs.Textbox(lines=5, placeholder="Input Text...")],
+    outputs=gr.outputs.Textbox()
+)
+iface.launch()

faiss_index/index.faiss

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc3b849e7e69be61f237fb33e681af1d1cd07b8fd2e43dcf7bb3a586e230efdc
+size 156112941

faiss_index/index.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fb155ba8e658607b4ea588b34200131d2488bd2716967634db2c80259fb4969c
+size 56496382

requirements.txt

@@ -0,0 +1,5 @@
+faiss-cpu
+langchain[llms]
+beautifulsoup4
+sentence-transformers
+transformers