GenAIDevTOProd commited on
Commit
7df6cd5
·
verified ·
1 Parent(s): c5597b8

Upload 4 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ gita_vector_index/index.faiss filter=lfs diff=lfs merge=lfs -text
app.py ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import gradio as gr
3
+ import openai
4
+ from getpass import getpass
5
+ import os
6
+ from langchain.vectorstores import FAISS
7
+ from langchain.embeddings import HuggingFaceEmbeddings
8
+
9
+ #OpenRouter API
10
+ openai.api_key = os.environ.get("OPENROUTER_API_KEY")
11
+ openai.api_base = "https://openrouter.ai/api/v1"
12
+
13
+ # Load vector DB
14
+ embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
15
+ db = FAISS.load_local("gita_vector_index", embedding_model, allow_dangerous_deserialization=True)
16
+ retriever = db.as_retriever()
17
+
18
+ # Chat function
19
+ def Talk2Gita(query, history):
20
+ try:
21
+ docs = retriever.get_relevant_documents(query)
22
+ context = "\n\n".join([doc.page_content for doc in docs[:2]])
23
+
24
+ response = openai.ChatCompletion.create(
25
+ model="mistralai/mistral-7b-instruct",
26
+ messages=[
27
+ {"role": "system", "content": "You are a helpful assistant answering questions from the Bhagavad Gita."},
28
+ {"role": "user", "content": f"""Context:
29
+ {context}
30
+
31
+ Question:
32
+ {query}"""}
33
+ ],
34
+ max_tokens=300,
35
+ )
36
+
37
+ return response["choices"][0]["message"]["content"]
38
+
39
+ except Exception as e:
40
+ return f" Error: {str(e)}"
41
+
42
+ # Launch Gradio app
43
+ gr.ChatInterface(Talk2Gita, title="Talk to the Bhagavad Gita").launch(share=True)
gita_vector_index/index.faiss ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ee0f1c02b097f795d35ab19d27417dbb74737adc69a751a2e2c3de8b4d04160
3
+ size 393261
gita_vector_index/index.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9803cea0ce7c011efb6d96160b78455b5b3d15d7dd1ef3e218579374a4fdf66
3
+ size 131710
requirements.txt ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ gradio
2
+ openai
3
+ langchain
4
+ huggingface-hub
5
+ sentence-transformers
6
+ faiss-cpu