Spaces:
Sleeping
Sleeping
Update server.py
Browse files
server.py
CHANGED
|
@@ -12,27 +12,21 @@ import nltk
|
|
| 12 |
from modal_tool import train_with_modal, app, upload_to_hf_from_volume, evaluate_model
|
| 13 |
|
| 14 |
|
| 15 |
-
GROQ_API_KEY = os.
|
| 16 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 17 |
|
| 18 |
if not GROQ_API_KEY:
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
try:
|
| 22 |
-
nltk.data.find('tokenizers/punkt')
|
| 23 |
-
except LookupError:
|
| 24 |
-
nltk.download('punkt', quiet=True)
|
| 25 |
-
|
| 26 |
-
mcp = FastMCP(name="FistalMCP")
|
| 27 |
|
| 28 |
GOOGLE_API_KEY_1 = os.environ.get("GOOGLE_API_KEY_1")
|
| 29 |
GOOGLE_API_KEY_2 = os.environ.get("GOOGLE_API_KEY_2")
|
| 30 |
GOOGLE_API_KEY_3 = os.environ.get("GOOGLE_API_KEY_3")
|
| 31 |
|
| 32 |
-
GOOGLE_API_KEYS = [
|
| 33 |
|
| 34 |
if not GOOGLE_API_KEYS:
|
| 35 |
-
|
|
|
|
| 36 |
|
| 37 |
|
| 38 |
async def genBatch(topic: str, samples_per_batch: int, batch_num: int, api_key: str, task_type: str) -> list:
|
|
|
|
| 12 |
from modal_tool import train_with_modal, app, upload_to_hf_from_volume, evaluate_model
|
| 13 |
|
| 14 |
|
| 15 |
+
GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
|
| 16 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 17 |
|
| 18 |
if not GROQ_API_KEY:
|
| 19 |
+
print("⚠ GROQ_API_KEY missing!")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
GOOGLE_API_KEY_1 = os.environ.get("GOOGLE_API_KEY_1")
|
| 22 |
GOOGLE_API_KEY_2 = os.environ.get("GOOGLE_API_KEY_2")
|
| 23 |
GOOGLE_API_KEY_3 = os.environ.get("GOOGLE_API_KEY_3")
|
| 24 |
|
| 25 |
+
GOOGLE_API_KEYS = [k for k in [GOOGLE_API_KEY_1, GOOGLE_API_KEY_2, GOOGLE_API_KEY_3] if k]
|
| 26 |
|
| 27 |
if not GOOGLE_API_KEYS:
|
| 28 |
+
print("⚠ No Google API keys found!")
|
| 29 |
+
|
| 30 |
|
| 31 |
|
| 32 |
async def genBatch(topic: str, samples_per_batch: int, batch_num: int, api_key: str, task_type: str) -> list:
|