ldhldh commited on
Commit
d7359cd
โ€ข
1 Parent(s): ab29248

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -36,6 +36,7 @@ print(f"๋ฌธ์„œ์— {len(texts)}๊ฐœ์˜ ๋ฌธ์„œ๋ฅผ ๊ฐ€์ง€๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค.")
36
 
37
  # ์ž„๋ฒ ๋”ฉ ๋ชจ๋ธ ๋กœ๋“œ
38
  embeddings = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-large")
 
39
  # ๋ฌธ์„œ์— ์žˆ๋Š” ํ…์ŠคํŠธ๋ฅผ ์ž„๋ฒ ๋”ฉํ•˜๊ณ  FAISS ์— ์ธ๋ฑ์Šค๋ฅผ ๊ตฌ์ถ•ํ•จ
40
  index = FAISS.from_documents(
41
  documents=texts,
@@ -44,6 +45,7 @@ index = FAISS.from_documents(
44
 
45
  # faiss_db ๋กœ ๋กœ์ปฌ์— ์ €์žฅํ•˜๊ธฐ
46
  index.save_local("")
 
47
  # faiss_db ๋กœ ๋กœ์ปฌ์— ๋กœ๋“œํ•˜๊ธฐ
48
  docsearch = FAISS.load_local("", embeddings)
49
 
@@ -138,7 +140,7 @@ def gen(x, id, customer_data):
138
  )
139
  query=f"๋‚˜๋Š” ํ˜„์žฌ {customer_data_list[index]}๋งŒ ๊ฐ€์ž…ํ•œ ์ƒํ™ฉ์ด์•ผ. {x}"
140
  response = qa({"query":query})
141
- output_str = response['result']
142
  print(prompt_template + output_str)
143
  history[index] += f"๊ณ ๊ฐ:{x}\n์ƒ๋‹ด์›:{output_str}\n"
144
  return output_str
 
36
 
37
  # ์ž„๋ฒ ๋”ฉ ๋ชจ๋ธ ๋กœ๋“œ
38
  embeddings = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-large")
39
+
40
  # ๋ฌธ์„œ์— ์žˆ๋Š” ํ…์ŠคํŠธ๋ฅผ ์ž„๋ฒ ๋”ฉํ•˜๊ณ  FAISS ์— ์ธ๋ฑ์Šค๋ฅผ ๊ตฌ์ถ•ํ•จ
41
  index = FAISS.from_documents(
42
  documents=texts,
 
45
 
46
  # faiss_db ๋กœ ๋กœ์ปฌ์— ์ €์žฅํ•˜๊ธฐ
47
  index.save_local("")
48
+
49
  # faiss_db ๋กœ ๋กœ์ปฌ์— ๋กœ๋“œํ•˜๊ธฐ
50
  docsearch = FAISS.load_local("", embeddings)
51
 
 
140
  )
141
  query=f"๋‚˜๋Š” ํ˜„์žฌ {customer_data_list[index]}๋งŒ ๊ฐ€์ž…ํ•œ ์ƒํ™ฉ์ด์•ผ. {x}"
142
  response = qa({"query":query})
143
+ output_str = response['result'].rsplit(".",1)[0]+"."
144
  print(prompt_template + output_str)
145
  history[index] += f"๊ณ ๊ฐ:{x}\n์ƒ๋‹ด์›:{output_str}\n"
146
  return output_str