Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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')
|