Update app.py

app.py

@@ -9,13 +9,15 @@ import pandas as pd
 from tqdm import tqdm
 import os
 from tensorflow.python.client import device_lib
+import safetensor
 
+PATH = './checkpoint-7500/'
 SEQ_LEN = 128
 tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
 
 def create_sentiment_bert():
   # 버트 pretrained 모델 로드
-  model = TFBertModel.from_pretrained('bert-base-multilingual-cased')
+  model = model = AutoModel.from_pretrained(PATH,local_files_only=True)
   # 토큰 인풋, 마스크 인풋, 세그먼트 인풋 정의
   token_inputs = tf.keras.layers.Input((SEQ_LEN,), dtype=tf.int32, name='input_word_ids')
   mask_inputs = tf.keras.layers.Input((SEQ_LEN,), dtype=tf.int32, name='input_masks')