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)
|