Spaces:
Running
on
Zero
Running
on
Zero
fix model-name
#21
by
reach-vb
HF Staff
- opened
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))
|