Kiy-K commited on
Commit
1b027ae
·
verified ·
1 Parent(s): c33d98d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -8,6 +8,11 @@ from datasets import load_dataset
8
  from duckduckgo_search import DDGS
9
  from transformers import pipeline
10
  import torch
 
 
 
 
 
11
 
12
  # ---------------- CONFIG ----------------
13
  MODEL_ID = "openai/gpt-oss-120b"
 
8
  from duckduckgo_search import DDGS
9
  from transformers import pipeline
10
  import torch
11
+ from cerebras.cloud.sdk import Cerebras
12
+
13
+ # Lấy token
14
+ CEREBRAS_TOKEN = os.getenv("CEREBRAS_API_KEY")
15
+ client = Cerebras(api_key=CEREBRAS_TOKEN)
16
 
17
  # ---------------- CONFIG ----------------
18
  MODEL_ID = "openai/gpt-oss-120b"