Spaces:
Sleeping
Sleeping
Final Compliance check: Added HF_TOKEN variable placeholder for Hugging Face Space submission.
Browse files- 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("
|
| 25 |
return
|
| 26 |
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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"]
|