Spaces:
Running
Running
Update app.py
Browse files
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:
|