Spaces:
Runtime error
Runtime error
Update model_functions.py
Browse files- model_functions.py +2 -3
model_functions.py
CHANGED
@@ -7,14 +7,13 @@ import streamlit as st
|
|
7 |
|
8 |
|
9 |
|
10 |
-
@st.cache(show_spinner=False)
|
11 |
def load_sentiment_analyzer():
|
12 |
tokenizer = AutoTokenizer.from_pretrained("aliciiavs/sentiment-analysis-whatsapp2")
|
13 |
model = AutoModelForSequenceClassification.from_pretrained("aliciiavs/sentiment-analysis-whatsapp2")
|
14 |
|
15 |
return tokenizer, model
|
16 |
|
17 |
-
|
18 |
def load_summarizer():
|
19 |
config = PeftConfig.from_pretrained("marcelomoreno26/bart-large-samsum-adapter")
|
20 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large")
|
@@ -25,7 +24,7 @@ def load_summarizer():
|
|
25 |
|
26 |
return tokenizer, model
|
27 |
|
28 |
-
|
29 |
def load_NER():
|
30 |
config = AutoConfig.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann")
|
31 |
model = AutoModelForTokenClassification.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann",config=config)
|
|
|
7 |
|
8 |
|
9 |
|
|
|
10 |
def load_sentiment_analyzer():
|
11 |
tokenizer = AutoTokenizer.from_pretrained("aliciiavs/sentiment-analysis-whatsapp2")
|
12 |
model = AutoModelForSequenceClassification.from_pretrained("aliciiavs/sentiment-analysis-whatsapp2")
|
13 |
|
14 |
return tokenizer, model
|
15 |
|
16 |
+
|
17 |
def load_summarizer():
|
18 |
config = PeftConfig.from_pretrained("marcelomoreno26/bart-large-samsum-adapter")
|
19 |
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large")
|
|
|
24 |
|
25 |
return tokenizer, model
|
26 |
|
27 |
+
|
28 |
def load_NER():
|
29 |
config = AutoConfig.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann")
|
30 |
model = AutoModelForTokenClassification.from_pretrained("hannahisrael03/wikineural-multilingual-ner-finetuned-wikiann",config=config)
|