MissingBreath commited on
Commit
65d6b85
·
verified ·
1 Parent(s): ab70f79

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -2
api.py CHANGED
@@ -43,8 +43,8 @@ import os
43
 
44
 
45
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
46
- MODEL_DIR = "/my_model"
47
- TOKENIZER_DIR = "/my_tokenizer"
48
  # Load the model and tokenizer
49
  try:
50
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_DIR)
 
43
 
44
 
45
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
46
+ MODEL_DIR = "./my_model"
47
+ TOKENIZER_DIR = "./my_tokenizer"
48
  # Load the model and tokenizer
49
  try:
50
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_DIR)