prithic07 commited on
Commit
2599a77
·
1 Parent(s): 4bc8817

Final Compliance check: Added HF_TOKEN variable placeholder for Hugging Face Space submission.

Browse files
Files changed (1) hide show
  1. inference.py +7 -2
inference.py CHANGED
@@ -18,13 +18,18 @@ load_dotenv()
18
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://generativelanguage.googleapis.com/v1beta/openai/")
19
  MODEL_NAME = os.environ.get("MODEL_NAME", "gemini-1.5-flash")
20
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", os.environ.get("GOOGLE_API_KEY", ""))
 
21
 
22
  def run_inference():
23
  if not GEMINI_API_KEY:
24
- print("ERROR: GEMINI_API_KEY not found.")
25
  return
26
 
27
- client = OpenAI(api_key=GEMINI_API_KEY, base_url=API_BASE_URL)
 
 
 
 
28
  env = ContextPruningEnv()
29
 
30
  tasks = ["noise_purge", "dedupe_arena", "signal_extract"]
 
18
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://generativelanguage.googleapis.com/v1beta/openai/")
19
  MODEL_NAME = os.environ.get("MODEL_NAME", "gemini-1.5-flash")
20
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", os.environ.get("GOOGLE_API_KEY", ""))
21
+ HF_TOKEN = os.environ.get("HF_TOKEN", "") # Required for HF Space compliance
22
 
23
  def run_inference():
24
  if not GEMINI_API_KEY:
25
+ print("Error: GEMINI_API_KEY or GOOGLE_API_KEY must be set.")
26
  return
27
 
28
+ # Initialize OpenAI client with Google's endpoint
29
+ client = OpenAI(
30
+ api_key=GEMINI_API_KEY,
31
+ base_url=API_BASE_URL
32
+ )
33
  env = ContextPruningEnv()
34
 
35
  tasks = ["noise_purge", "dedupe_arena", "signal_extract"]