Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,11 @@ from langchain_community.llms import HuggingFacePipeline
|
|
8 |
|
9 |
#Konstanten
|
10 |
ANTI_BOT_PW = os.getenv("CORRECT_VALIDATE")
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
# Hugging Face Token direkt im Code setzen
|
13 |
hf_token = os.getenv("HF_READ")
|
@@ -16,8 +21,8 @@ hf_token = os.getenv("HF_READ")
|
|
16 |
embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
|
17 |
|
18 |
# Initialisierung von Tokenizer und RAG Modell mit Token
|
19 |
-
tokenizer = RagTokenizer.from_pretrained("facebook/rag-sequence-nq",
|
20 |
-
model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-nq",
|
21 |
|
22 |
# Verbindung zur Chroma DB und Laden der Dokumente
|
23 |
chroma_db = Chroma(embedding_model=embedding_model, persist_directory = PATH_WORK + CHROMA_DIR)
|
|
|
8 |
|
9 |
#Konstanten
|
10 |
ANTI_BOT_PW = os.getenv("CORRECT_VALIDATE")
|
11 |
+
PATH_WORK = "."
|
12 |
+
CHROMA_DIR = "/chroma/kkg"
|
13 |
+
CHROMA_PDF = './chroma/kkg/pdf'
|
14 |
+
CHROMA_WORD = './chroma/kkg/word'
|
15 |
+
CHROMA_EXCEL = './chroma/kkg/excel'
|
16 |
|
17 |
# Hugging Face Token direkt im Code setzen
|
18 |
hf_token = os.getenv("HF_READ")
|
|
|
21 |
embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
|
22 |
|
23 |
# Initialisierung von Tokenizer und RAG Modell mit Token
|
24 |
+
tokenizer = RagTokenizer.from_pretrained("facebook/rag-sequence-nq", token=hf_token)
|
25 |
+
model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-nq", token=hf_token)
|
26 |
|
27 |
# Verbindung zur Chroma DB und Laden der Dokumente
|
28 |
chroma_db = Chroma(embedding_model=embedding_model, persist_directory = PATH_WORK + CHROMA_DIR)
|