Spaces:
Sleeping
Sleeping
Runnies23
commited on
Commit
·
861b48c
1
Parent(s):
256f405
Add application file
Browse files
app.py
CHANGED
@@ -20,9 +20,11 @@ import sqlite3
|
|
20 |
def load_model():
|
21 |
return BGEM3FlagModel('BAAI/bge-m3',use_fp16=True)
|
22 |
|
|
|
23 |
def load_rerank_model():
|
24 |
return FlagReranker('BAAI/bge-reranker-v2-m3', use_fp16=True)
|
25 |
|
|
|
26 |
def initLLM():
|
27 |
os.environ["GOOGLE_API_KEY"] = "AIzaSyAuKPswmbdM8jCpSt0luez7tjLND-uyY7M"
|
28 |
llm = ChatGoogleGenerativeAI(model="gemini-pro")
|
@@ -40,12 +42,15 @@ def initLLM():
|
|
40 |
)
|
41 |
llm_chain = LLMChain(prompt=prompt, llm=llm)
|
42 |
return llm_chain
|
|
|
|
|
43 |
def embeded_corpus():
|
44 |
file_path_embeded_corpus = "save/BGM3savesimilar_Corpus" #
|
45 |
with open(file_path_embeded_corpus,'rb') as file :
|
46 |
BGM3similar_Corpus = pickle.load(file)
|
47 |
return BGM3similar_Corpus
|
48 |
|
|
|
49 |
def insert_feedback(question, answer,like,dislike, feedback_text):
|
50 |
conn = sqlite3.connect('feedback.db')
|
51 |
cursor = conn.cursor()
|
|
|
20 |
def load_model():
|
21 |
return BGEM3FlagModel('BAAI/bge-m3',use_fp16=True)
|
22 |
|
23 |
+
@st.cache_resource
|
24 |
def load_rerank_model():
|
25 |
return FlagReranker('BAAI/bge-reranker-v2-m3', use_fp16=True)
|
26 |
|
27 |
+
@st.cache_resource
|
28 |
def initLLM():
|
29 |
os.environ["GOOGLE_API_KEY"] = "AIzaSyAuKPswmbdM8jCpSt0luez7tjLND-uyY7M"
|
30 |
llm = ChatGoogleGenerativeAI(model="gemini-pro")
|
|
|
42 |
)
|
43 |
llm_chain = LLMChain(prompt=prompt, llm=llm)
|
44 |
return llm_chain
|
45 |
+
|
46 |
+
@st.cache_resource
|
47 |
def embeded_corpus():
|
48 |
file_path_embeded_corpus = "save/BGM3savesimilar_Corpus" #
|
49 |
with open(file_path_embeded_corpus,'rb') as file :
|
50 |
BGM3similar_Corpus = pickle.load(file)
|
51 |
return BGM3similar_Corpus
|
52 |
|
53 |
+
@st.cache_resource
|
54 |
def insert_feedback(question, answer,like,dislike, feedback_text):
|
55 |
conn = sqlite3.connect('feedback.db')
|
56 |
cursor = conn.cursor()
|