dkundel-openai reach-vb HF Staff commited on
Commit
62686fb
·
verified ·
1 Parent(s): eb6b4ca

fix model-name (#21)

Browse files

- fix model-name (48bf555175a98b65fa19cfb04181141027dd1ee5)


Co-authored-by: Vaibhav Srivastav <reach-vb@users.noreply.huggingface.co>

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", "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))
 
7
  import spaces
8
 
9
  # === Config (override via Space secrets/env vars) ===
10
+ MODEL_ID = os.environ.get("MODEL_ID", "openai/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))