sofzcc commited on
Commit
1518512
·
verified ·
1 Parent(s): 4c49fe0

Update app/config.py

Browse files
Files changed (1) hide show
  1. app/config.py +20 -0
app/config.py CHANGED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from pydantic_settings import BaseSettings
2
+
3
+
4
+ class Settings(BaseSettings):
5
+
6
+ model_name: str = "microsoft/Phi-3-mini-4k-instruct"
7
+
8
+ embedding_model: str = (
9
+ "sentence-transformers/all-MiniLM-L6-v2"
10
+ )
11
+
12
+ device: str = "cuda"
13
+
14
+ top_k: int = 5
15
+
16
+ class Config:
17
+ env_file = ".env"
18
+
19
+
20
+ settings = Settings()