Commit ·
e775096
1
Parent(s): f4c5d7f
Use HF Nemotron Mini for local validation
Browse files- gradio_pharmacopilot_demo.py +1 -1
- requirements.txt +0 -1
gradio_pharmacopilot_demo.py
CHANGED
|
@@ -52,7 +52,7 @@ INVENTORY_PATH = data_path("inventory.json")
|
|
| 52 |
MODEL_ID = os.getenv("PHARMACOPILOT_MODEL_ID", "openbmb/MiniCPM-V-4_5")
|
| 53 |
LIVE_GPU_OCR = os.getenv("PHARMACOPILOT_LIVE_GPU_OCR", "1").lower() not in {"0", "false", "no"}
|
| 54 |
LIVE_NEMOTRON = os.getenv("PHARMACOPILOT_LIVE_NEMOTRON", "1").lower() not in {"0", "false", "no"}
|
| 55 |
-
NEMOTRON_MODEL_ID = os.getenv("NEMOTRON_MODEL_ID", "nvidia/
|
| 56 |
NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY", "")
|
| 57 |
NVIDIA_BASE_URL = os.getenv("NVIDIA_BASE_URL", "https://integrate.api.nvidia.com/v1")
|
| 58 |
NVIDIA_NIM_MODEL = os.getenv("NVIDIA_NIM_MODEL", "nvidia/nvidia-nemotron-nano-9b-v2")
|
|
|
|
| 52 |
MODEL_ID = os.getenv("PHARMACOPILOT_MODEL_ID", "openbmb/MiniCPM-V-4_5")
|
| 53 |
LIVE_GPU_OCR = os.getenv("PHARMACOPILOT_LIVE_GPU_OCR", "1").lower() not in {"0", "false", "no"}
|
| 54 |
LIVE_NEMOTRON = os.getenv("PHARMACOPILOT_LIVE_NEMOTRON", "1").lower() not in {"0", "false", "no"}
|
| 55 |
+
NEMOTRON_MODEL_ID = os.getenv("NEMOTRON_MODEL_ID", "nvidia/Nemotron-Mini-4B-Instruct")
|
| 56 |
NVIDIA_API_KEY = os.getenv("NVIDIA_API_KEY", "")
|
| 57 |
NVIDIA_BASE_URL = os.getenv("NVIDIA_BASE_URL", "https://integrate.api.nvidia.com/v1")
|
| 58 |
NVIDIA_NIM_MODEL = os.getenv("NVIDIA_NIM_MODEL", "nvidia/nvidia-nemotron-nano-9b-v2")
|
requirements.txt
CHANGED
|
@@ -10,4 +10,3 @@ sentencepiece
|
|
| 10 |
protobuf
|
| 11 |
einops
|
| 12 |
timm
|
| 13 |
-
openai>=1.88.0
|
|
|
|
| 10 |
protobuf
|
| 11 |
einops
|
| 12 |
timm
|
|
|