duwing commited on
Commit
20491a1
ยท
verified ยท
1 Parent(s): 1cf26cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
17
 
18
  def create_sentiment_bert():
19
  # ๋ฒ„ํŠธ pretrained ๋ชจ๋ธ ๋กœ๋“œ
20
- model = model = AutoModel.from_pretrained(PATH,local_files_only=True)
21
  # ํ† ํฐ ์ธํ’‹, ๋งˆ์Šคํฌ ์ธํ’‹, ์„ธ๊ทธ๋จผํŠธ ์ธํ’‹ ์ •์˜
22
  token_inputs = tf.keras.layers.Input((SEQ_LEN,), dtype=tf.int32, name='input_word_ids')
23
  mask_inputs = tf.keras.layers.Input((SEQ_LEN,), dtype=tf.int32, name='input_masks')
 
17
 
18
  def create_sentiment_bert():
19
  # ๋ฒ„ํŠธ pretrained ๋ชจ๋ธ ๋กœ๋“œ
20
+ model = model = TFAutoModel.from_pretrained(PATH,local_files_only=True)
21
  # ํ† ํฐ ์ธํ’‹, ๋งˆ์Šคํฌ ์ธํ’‹, ์„ธ๊ทธ๋จผํŠธ ์ธํ’‹ ์ •์˜
22
  token_inputs = tf.keras.layers.Input((SEQ_LEN,), dtype=tf.int32, name='input_word_ids')
23
  mask_inputs = tf.keras.layers.Input((SEQ_LEN,), dtype=tf.int32, name='input_masks')