Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,7 +17,6 @@ st.title("Blah-1")
|
|
| 17 |
|
| 18 |
# ----------------- API Keys -----------------
|
| 19 |
os.environ["GROQ_API_KEY"] = st.secrets.get("GROQ_API_KEY", "")
|
| 20 |
-
os.environ["HF_TOKEN"] = st.secrets.get("HF_TOKEN", "")
|
| 21 |
|
| 22 |
# ----------------- Clear ChromaDB Cache -----------------
|
| 23 |
chromadb.api.client.SharedSystemClient.clear_system_cache()
|
|
@@ -42,7 +41,7 @@ rag_llm = ChatGroq(model="mixtral-8x7b-32768")
|
|
| 42 |
llm_judge.verbose = True
|
| 43 |
rag_llm.verbose = True
|
| 44 |
|
| 45 |
-
# ----------------- PDF Selection
|
| 46 |
st.subheader("π PDF Selection")
|
| 47 |
pdf_source = st.radio("Choose a PDF source:", ["Upload a PDF file", "Enter a PDF URL"], index=0, horizontal=True)
|
| 48 |
|
|
@@ -82,7 +81,7 @@ if not st.session_state.pdf_loaded and "pdf_path" in st.session_state:
|
|
| 82 |
docs = loader.load()
|
| 83 |
st.json(docs[0].metadata)
|
| 84 |
|
| 85 |
-
# Embedding Model
|
| 86 |
model_name = "nomic-ai/modernbert-embed-base"
|
| 87 |
embedding_model = HuggingFaceEmbeddings(model_name=model_name, model_kwargs={"device": "cpu"})
|
| 88 |
|
|
|
|
| 17 |
|
| 18 |
# ----------------- API Keys -----------------
|
| 19 |
os.environ["GROQ_API_KEY"] = st.secrets.get("GROQ_API_KEY", "")
|
|
|
|
| 20 |
|
| 21 |
# ----------------- Clear ChromaDB Cache -----------------
|
| 22 |
chromadb.api.client.SharedSystemClient.clear_system_cache()
|
|
|
|
| 41 |
llm_judge.verbose = True
|
| 42 |
rag_llm.verbose = True
|
| 43 |
|
| 44 |
+
# ----------------- PDF Selection -----------------
|
| 45 |
st.subheader("π PDF Selection")
|
| 46 |
pdf_source = st.radio("Choose a PDF source:", ["Upload a PDF file", "Enter a PDF URL"], index=0, horizontal=True)
|
| 47 |
|
|
|
|
| 81 |
docs = loader.load()
|
| 82 |
st.json(docs[0].metadata)
|
| 83 |
|
| 84 |
+
# Embedding Model
|
| 85 |
model_name = "nomic-ai/modernbert-embed-base"
|
| 86 |
embedding_model = HuggingFaceEmbeddings(model_name=model_name, model_kwargs={"device": "cpu"})
|
| 87 |
|