Spaces:
Running
Running
Jin Zhu
commited on
Commit
·
2af30bb
1
Parent(s):
ecfa074
Update app.py
Browse files- src/app.py +1 -1
src/app.py
CHANGED
|
@@ -85,7 +85,7 @@ def load_model(from_pretrained, base_model, cache_dir, device):
|
|
| 85 |
# Configuration
|
| 86 |
# -----------------
|
| 87 |
MODEL_CONFIG = {
|
| 88 |
-
'from_pretrained': 'mamba413/AdaDetectGPT-Model',
|
| 89 |
'base_model': 'gemma-1b',
|
| 90 |
'cache_dir': '../cache',
|
| 91 |
'device': 'mps',
|
|
|
|
| 85 |
# Configuration
|
| 86 |
# -----------------
|
| 87 |
MODEL_CONFIG = {
|
| 88 |
+
'from_pretrained': 'mamba413/AdaDetectGPT-Model' if os.environ.get('SPACE_ID') else './FineTune/ckpt/lr_0.0001_a_1',
|
| 89 |
'base_model': 'gemma-1b',
|
| 90 |
'cache_dir': '../cache',
|
| 91 |
'device': 'mps',
|