helloparthshah commited on
Commit
958b115
·
1 Parent(s): d257e9f

Updated requirements

Browse files
requirements.txt CHANGED
@@ -14,5 +14,7 @@ Requests==2.32.3
14
  sentence_transformers==4.1.0
15
  starlette==0.46.2
16
  uvicorn==0.34.2
 
 
17
  --extra-index-url https://download.pytorch.org/whl/cu121
18
  torch==2.5.1+cu121
 
14
  sentence_transformers==4.1.0
15
  starlette==0.46.2
16
  uvicorn==0.34.2
17
+ google-genai==1.17.0
18
+ itsdangerous==2.2.0
19
  --extra-index-url https://download.pytorch.org/whl/cu121
20
  torch==2.5.1+cu121
src/manager/utils/runtime_selector.py DELETED
@@ -1,14 +0,0 @@
1
- import platform
2
- import GPUtil
3
- from dotenv import load_dotenv
4
- load_dotenv()
5
-
6
- def detect_runtime_environment():
7
- os_type = platform.system().lower()
8
- gpus = GPUtil.getGPUs()
9
- if gpus:
10
- return "gpu"
11
- elif os_type in ["darwin", "linux"]:
12
- return "cpu-local"
13
- else:
14
- return "cloud-only"