alexkueck commited on
Commit
6658f37
1 Parent(s): c6a49dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
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", use_auth_token=hf_token)
20
- model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-nq", use_auth_token=hf_token)
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)