fix model-name

#21
by reach-vb HF Staff - opened
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))