Nerva5678 commited on
Commit
171ee44
·
verified ·
1 Parent(s): 55bf44a

Upload 3 files

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -4,12 +4,14 @@ import torch
4
  import os
5
  import time
6
  import logging
7
- from langchain.embeddings.huggingface import HuggingFaceEmbeddings
8
  from langchain.vectorstores import FAISS
9
  from langchain.llms import HuggingFacePipeline
10
  from langchain.chains import RetrievalQA, LLMChain
11
  from langchain.prompts import PromptTemplate
12
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 
13
 
14
  # 設定logging
15
  logging.basicConfig(level=logging.INFO)
 
4
  import os
5
  import time
6
  import logging
7
+ from langchain_community.embeddings.huggingface import HuggingFaceEmbeddings
8
  from langchain.vectorstores import FAISS
9
  from langchain.llms import HuggingFacePipeline
10
  from langchain.chains import RetrievalQA, LLMChain
11
  from langchain.prompts import PromptTemplate
12
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
13
+ from langchain_community.vectorstores import FAISS
14
+ from langchain_community.llms import HuggingFacePipeline
15
 
16
  # 設定logging
17
  logging.basicConfig(level=logging.INFO)
requirements.txt CHANGED
@@ -5,5 +5,6 @@ transformers
5
  sentence-transformers
6
  faiss-cpu
7
  langchain
 
8
  protobuf
9
  openpyxl
 
5
  sentence-transformers
6
  faiss-cpu
7
  langchain
8
+ langchain-community
9
  protobuf
10
  openpyxl