Hugging Face's logo

Spaces:
planetearth79
/
ysda_hw
Sleeping

App Files Community
planetearth79 commited on
Commit
a008ff9
•
1 Parent(s): f22dfff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ text = title_text + " " + summary_text
49
 
50
 
51
  # 1
52
- @st.cache_resource
53
  def load_first_model():
54
  loaded_tokenizer = AutoTokenizer.from_pretrained("multi_class_model")
55
  loaded_model = AutoModelForSequenceClassification.from_pretrained("multi_class_model")
@@ -79,7 +79,7 @@ st.bar_chart(chart_data)
79
 
80
 
81
  # 2
82
- @st.cache_resource
83
  def load_first_model():
84
  loaded_tokenizer = AutoTokenizer.from_pretrained("multi_label_model")
85
  loaded_model = AutoModelForSequenceClassification.from_pretrained("multi_label_model")
 
49
 
50
 
51
  # 1
52
+ @st.cache
53
  def load_first_model():
54
  loaded_tokenizer = AutoTokenizer.from_pretrained("multi_class_model")
55
  loaded_model = AutoModelForSequenceClassification.from_pretrained("multi_class_model")
 
79
 
80
 
81
  # 2
82
+ @st.cache
83
  def load_first_model():
84
  loaded_tokenizer = AutoTokenizer.from_pretrained("multi_label_model")
85
  loaded_model = AutoModelForSequenceClassification.from_pretrained("multi_label_model")