reach-vb HF Staff commited on
Commit
9b81a05
·
verified ·
1 Parent(s): 3f5fdf1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import pipeline
7
  import spaces
8
 
9
  # === Config (override via Space secrets/env vars) ===
10
- MODEL_ID = os.environ.get("MODEL_ID", "tlhv/osb-minier")
11
  DEFAULT_MAX_NEW_TOKENS = int(os.environ.get("MAX_NEW_TOKENS", 512))
12
  DEFAULT_TEMPERATURE = float(os.environ.get("TEMPERATURE", 1))
13
  DEFAULT_TOP_P = float(os.environ.get("TOP_P", 1.0))
 
7
  import spaces
8
 
9
  # === Config (override via Space secrets/env vars) ===
10
+ MODEL_ID = os.environ.get("MODEL_ID", "gpt-oss-safeguard-20b")
11
  DEFAULT_MAX_NEW_TOKENS = int(os.environ.get("MAX_NEW_TOKENS", 512))
12
  DEFAULT_TEMPERATURE = float(os.environ.get("TEMPERATURE", 1))
13
  DEFAULT_TOP_P = float(os.environ.get("TOP_P", 1.0))