DrishtiSharma commited on
Commit
a601e7b
Β·
verified Β·
1 Parent(s): 5ea6ce5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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 (Upload or URL) -----------------
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 (HF on CPU)
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