Chirag0123 commited on
Commit
35f712a
Β·
1 Parent(s): e40cf69

fix(phase2): Update inference script and requirements for rigorous OpenEnv Phase 2 hackathon compliance (HF_TOKEN exception, valid defaults, dependencies)

Browse files
Files changed (2) hide show
  1. inference.py +4 -2
  2. requirements.txt +1 -0
inference.py CHANGED
@@ -17,9 +17,11 @@ from openai import OpenAI
17
  import httpx
18
 
19
  # ── Configuration ─────────────────────────────────────────────────────────────
20
- API_BASE_URL = os.getenv("API_BASE_URL", "<your-active-endpoint>")
21
- MODEL_NAME = os.getenv("MODEL_NAME", "<your-active-model>")
22
  HF_TOKEN = os.getenv("HF_TOKEN")
 
 
23
 
24
  # Optional β€” if you use from_docker_image():
25
  LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME")
 
17
  import httpx
18
 
19
  # ── Configuration ─────────────────────────────────────────────────────────────
20
+ API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
21
+ MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4-turbo-preview")
22
  HF_TOKEN = os.getenv("HF_TOKEN")
23
+ if not HF_TOKEN:
24
+ raise ValueError("HF_TOKEN environment variable is required but missing.")
25
 
26
  # Optional β€” if you use from_docker_image():
27
  LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME")
requirements.txt CHANGED
@@ -6,3 +6,4 @@ httpx
6
  pytest
7
  gradio>=4.0
8
  huggingface_hub
 
 
6
  pytest
7
  gradio>=4.0
8
  huggingface_hub
9
+ openenv-core>=0.2.0