rajistics commited on
Commit
2e6b9d1
1 Parent(s): 46780c2

Removed cache

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import psutil
8
  import streamlit as st
9
  from PIL import Image
10
  from streamlit import components
11
- from streamlit.caching import clear_cache
12
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
13
  from transformers_interpret import SequenceClassificationExplainer
14
 
@@ -22,7 +22,7 @@ def print_memory_usage():
22
  logging.info(f"RAM memory % used: {psutil.virtual_memory()[2]}")
23
 
24
 
25
- @st.cache(allow_output_mutation=True, suppress_st_warning=True, max_entries=1)
26
  def load_model(model_name):
27
  return (
28
  AutoModelForSequenceClassification.from_pretrained(model_name),
 
8
  import streamlit as st
9
  from PIL import Image
10
  from streamlit import components
11
+ #from streamlit.caching import clear_cache
12
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
13
  from transformers_interpret import SequenceClassificationExplainer
14
 
 
22
  logging.info(f"RAM memory % used: {psutil.virtual_memory()[2]}")
23
 
24
 
25
+ #@st.cache(allow_output_mutation=True, suppress_st_warning=True, max_entries=1)
26
  def load_model(model_name):
27
  return (
28
  AutoModelForSequenceClassification.from_pretrained(model_name),