BivinSadler commited on
Commit
4021eba
Β·
verified Β·
1 Parent(s): 506c657

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +6 -2
  2. requirements.txt +9 -7
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  # βœ… Set up your OpenAI API Key
2
  import os
3
  os.environ["OPENAI_API_KEY"] = "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" # πŸ” Replace with your key
@@ -25,8 +29,8 @@ def build_rag_agent(pdf_path):
25
  return RetrievalQA.from_chain_type(llm=llm, retriever=retriever, chain_type="stuff")
26
 
27
  # βœ… Create RAG agents for course syllabi
28
- stat6371_agent = build_rag_agent("ds6371syllabusVer7.pdf")
29
- ds6306_agent = build_rag_agent("DDSSyllabus2025.pdf")
30
 
31
  # βœ… Load Hugging Face fine-tuned model for general statistics questions
32
  general_stat_agent = pipeline("text2text-generation", model="google/flan-t5-base") # Replace with your model
 
1
+ # βœ… Mount Google Drive
2
+ from google.colab import drive
3
+ drive.mount('/content/drive')
4
+
5
  # βœ… Set up your OpenAI API Key
6
  import os
7
  os.environ["OPENAI_API_KEY"] = "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" # πŸ” Replace with your key
 
29
  return RetrievalQA.from_chain_type(llm=llm, retriever=retriever, chain_type="stuff")
30
 
31
  # βœ… Create RAG agents for course syllabi
32
+ stat6371_agent = build_rag_agent("/content/drive/MyDrive/PDFs/ds_6371_syllabus Ver 7.pdf")
33
+ ds6306_agent = build_rag_agent("/content/drive/MyDrive/PDFs/DDS_Syllabus_2025.pdf")
34
 
35
  # βœ… Load Hugging Face fine-tuned model for general statistics questions
36
  general_stat_agent = pipeline("text2text-generation", model="google/flan-t5-base") # Replace with your model
requirements.txt CHANGED
@@ -1,9 +1,11 @@
1
- openai
2
- langchain
3
- langchain-community
4
- faiss-cpu
5
- PyPDF2
6
- transformers
7
  pypdf
 
8
  gradio
9
- tiktoken
 
 
1
+ openai
2
+ langchain
3
+ langchain-community
4
+ langchain-openai
5
+ faiss-cpu
6
+ PyPDF2
7
  pypdf
8
+ transformers
9
  gradio
10
+ tiktoken
11
+ torch # βœ… Required for HF pipeline