YO-LOLO commited on
Commit
29dcca7
·
verified ·
1 Parent(s): 8bc4508

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -19,6 +19,9 @@ from langchain.llms import HuggingFacePipeline
19
  from langchain.schema import Document
20
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
21
  from transformers import T5ForConditionalGeneration, T5Tokenizer
 
 
 
22
 
23
  # 条件付きインポート(ローカル環境とHugging Face Spacesの両方に対応)
24
  try:
 
19
  from langchain.schema import Document
20
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
21
  from transformers import T5ForConditionalGeneration, T5Tokenizer
22
+ from sentence_transformers import SentenceTransformer
23
+
24
+ model = SentenceTransformer("sentence-transformers/all-mpnet-base-v2") # 1024次元のモデル
25
 
26
  # 条件付きインポート(ローカル環境とHugging Face Spacesの両方に対応)
27
  try: