mahreenfathima commited on
Commit
14f5c79
·
verified ·
1 Parent(s): 9d0fa79

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +5 -11
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.getenv("GROQ_API_KEY")
16
  HF_TOKEN = os.environ.get("HF_TOKEN")
17
 
18
  if not GROQ_API_KEY:
19
- raise ValueError("HF secrets problem/")
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 = [key for key in [GOOGLE_API_KEY_1, GOOGLE_API_KEY_2, GOOGLE_API_KEY_3] if key]
33
 
34
  if not GOOGLE_API_KEYS:
35
- raise ValueError("Where are your keys?")
 
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: