Spaces:
Sleeping
Sleeping
ShobhitKori
commited on
Commit
·
e94ab80
1
Parent(s):
17fdc74
Updated codet5_model.py
Browse files- codet5_model.py +2 -1
codet5_model.py
CHANGED
|
@@ -37,8 +37,9 @@
|
|
| 37 |
import os
|
| 38 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 39 |
|
| 40 |
-
MODEL_DIR = "
|
| 41 |
|
|
|
|
| 42 |
# Global variables but not initialized
|
| 43 |
tokenizer = None
|
| 44 |
model = None
|
|
|
|
| 37 |
import os
|
| 38 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 39 |
|
| 40 |
+
MODEL_DIR = "./data/codet5-model"
|
| 41 |
|
| 42 |
+
os.makedirs(MODEL_DIR, exist_ok=True)
|
| 43 |
# Global variables but not initialized
|
| 44 |
tokenizer = None
|
| 45 |
model = None
|