mgbam commited on
Commit
834aa30
·
verified ·
1 Parent(s): 3e78ff5

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +18 -9
config.py CHANGED
@@ -1,18 +1,27 @@
1
  import os
2
 
3
- # API Keys
 
 
 
 
4
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
 
 
5
  PUBMED_EMAIL = os.getenv("PUBMED_EMAIL", "[email protected]")
6
 
7
- # OpenAI model
8
- OPENAI_DEFAULT_MODEL = "gpt-4o-mini"
9
 
10
- # Hugging Face Vision-Language Model
11
  HF_IMAGE_MODEL = "HuggingFaceTB/SmolVLM-500M-Instruct"
12
 
13
- # PubMed defaults
14
- PUBMED_MAX_RESULTS = 5
15
- CHUNK_SIZE = 512
 
 
 
16
 
17
- # Application-wide settings
18
- APP_TITLE = "Next-Gen Medical AI"
 
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