marcelomoreno26 commited on
Commit
8653d5d
1 Parent(s): a6f40d0

Update model_functions.py

Browse files
Files changed (1) hide show
  1. 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
- @st.cache(show_spinner=False)
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
- @st.cache(show_spinner=False)
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)