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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -1,7 +1,3 @@
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,8 +25,8 @@ def build_rag_agent(pdf_path):
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
 
 
 
 
 
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
  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