mgbam commited on
Commit
504e4af
·
verified ·
1 Parent(s): 95350ad

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +9 -20
config.py CHANGED
@@ -1,27 +1,16 @@
1
  import os
2
 
3
- ###############################################################################
4
- # ENVIRONMENT & KEY MANAGEMENT #
5
- ###############################################################################
 
6
 
7
- # OpenAI API Key
8
- OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
9
 
10
- # PubMed / NCBI usage
11
- PUBMED_EMAIL = os.getenv("PUBMED_EMAIL", "[email protected]")
12
-
13
- # Default LLM Model
14
- OPENAI_DEFAULT_MODEL = "gpt-4o-mini" # Example; replace with real name if needed
15
-
16
- # Hugging Face Image-Text Model for advanced medical imaging
17
  HF_IMAGE_MODEL = "HuggingFaceTB/SmolVLM-500M-Instruct"
18
 
19
- # Vector DB paths/config
20
- VECTORDB_PATH = "vector_db"
21
- EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
22
-
23
- # Summaries
24
- DEFAULT_SUMMARIZATION_CHUNK = 512
25
-
26
- # Misc settings
27
  MAX_PUBMED_RESULTS = 5
 
 
1
  import os
2
 
3
+ # Secrets and environment variables
4
+ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
5
+ MY_PUBMED_EMAIL = os.getenv("MY_PUBMED_EMAIL")
6
+ HF_TOKEN = os.getenv("HF_TOKEN")
7
 
8
+ # Default OpenAI model
9
+ OPENAI_DEFAULT_MODEL = "gpt-4o-mini"
10
 
11
+ # Hugging Face model for medical image recognition
 
 
 
 
 
 
12
  HF_IMAGE_MODEL = "HuggingFaceTB/SmolVLM-500M-Instruct"
13
 
14
+ # PubMed and summarization settings
 
 
 
 
 
 
 
15
  MAX_PUBMED_RESULTS = 5
16
+ SUMMARIZATION_CHUNK_SIZE = 512