Jaamie commited on
Commit
d30ff12
·
verified ·
1 Parent(s): 2431ab4

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -44,7 +44,7 @@ resend_api_key = os.getenv("RESEND_API_KEY")
44
  login(token=hf_token)
45
 
46
  # Mount Google Drive
47
- drive.mount('/content/drive')
48
 
49
  # -------------------------------
50
  # 🔧 Configuration
@@ -57,7 +57,8 @@ embedding_model_bge = "BAAI/bge-base-en-v1.5"
57
  #save_path_bge = "./models/bge-base-en-v1.5"
58
  faiss_index_path = "./qa_faiss_embedding.index"
59
  chunked_text_path = "./chunked_text_RAG_text.txt"
60
- READER_MODEL_NAME = "google/gemma-2-9b-it"
 
61
  log_file_path = "./diagnosis_logs.csv"
62
  feedback_file_path = "./feedback_logs.csv"
63
 
@@ -82,7 +83,7 @@ if not os.path.exists(feedback_file_path):
82
  ])
83
 
84
  # Ensure directory exists
85
- os.makedirs(save_path_bge, exist_ok=True)
86
 
87
  # -------------------------------
88
  # 🔧 Model setup
 
44
  login(token=hf_token)
45
 
46
  # Mount Google Drive
47
+ #drive.mount('/content/drive')
48
 
49
  # -------------------------------
50
  # 🔧 Configuration
 
57
  #save_path_bge = "./models/bge-base-en-v1.5"
58
  faiss_index_path = "./qa_faiss_embedding.index"
59
  chunked_text_path = "./chunked_text_RAG_text.txt"
60
+ #READER_MODEL_NAME = "google/gemma-2-9b-it"
61
+ READER_MODEL_NAME = "google/gemma-2b-it"
62
  log_file_path = "./diagnosis_logs.csv"
63
  feedback_file_path = "./feedback_logs.csv"
64
 
 
83
  ])
84
 
85
  # Ensure directory exists
86
+ #os.makedirs(save_path_bge, exist_ok=True)
87
 
88
  # -------------------------------
89
  # 🔧 Model setup