Spaces:
Running
Running
Jin Zhu
commited on
Commit
·
ecfa074
1
Parent(s):
a9c7cad
add checkpoint file
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':
|
| 89 |
'base_model': 'gemma-1b',
|
| 90 |
'cache_dir': '../cache',
|
| 91 |
'device': 'mps',
|
|
|
|
| 85 |
# Configuration
|
| 86 |
# -----------------
|
| 87 |
MODEL_CONFIG = {
|
| 88 |
+
'from_pretrained': 'mamba413/AdaDetectGPT-Model',
|
| 89 |
'base_model': 'gemma-1b',
|
| 90 |
'cache_dir': '../cache',
|
| 91 |
'device': 'mps',
|