Emil25 commited on
Commit
6e4bfc9
·
verified ·
1 Parent(s): 6b7f048

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,6 +3,7 @@ import pandas as pd
3
  import numpy as np
4
  import nltk
5
  nltk.download('punkt')
 
6
  from nltk.tokenize import sent_tokenize, word_tokenize
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
@@ -15,7 +16,7 @@ st.set_page_config(
15
 
16
 
17
  # Загрузка модели и токенизатора
18
- @st.cache_resource()
19
  def get_model():
20
  # Загрузка модели
21
  model = AutoModelForCausalLM.from_pretrained('model')
 
3
  import numpy as np
4
  import nltk
5
  nltk.download('punkt')
6
+ nltk.download('punkt_tab')
7
  from nltk.tokenize import sent_tokenize, word_tokenize
8
  from transformers import AutoModelForCausalLM, AutoTokenizer
9
 
 
16
 
17
 
18
  # Загрузка модели и токенизатора
19
+ @st.cache_data()
20
  def get_model():
21
  # Загрузка модели
22
  model = AutoModelForCausalLM.from_pretrained('model')