Mohibullah commited on
Commit
e775096
·
1 Parent(s): f4c5d7f

Use HF Nemotron Mini for local validation

Browse files
Files changed (2) hide show
  1. gradio_pharmacopilot_demo.py +1 -1
  2. 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/NVIDIA-Nemotron-Nano-9B-v2")
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