Spaces:
Runtime error
Runtime error
Removed cache
Browse files
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 |
-
|
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),
|