Runnies23 commited on
Commit
861b48c
1 Parent(s): 256f405

Add application file

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