Hugging Face's logo

Spaces:
Seetha
/
IMA-pipeline-streamlit
Sleeping

App Files Community
2
Seetha commited on
Commit
3c8b9ba
•
1 Parent(s): ccad741

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -142,7 +142,7 @@ def main():
142
  st.write("--- %s seconds ---" % (time.time() - start_time))
143
 
144
  model_name = "distilbert-base-cased"
145
- tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
146
 
147
  model_path1 = "DistilBertforTokenclassification"
148
 
 
142
  st.write("--- %s seconds ---" % (time.time() - start_time))
143
 
144
  model_name = "distilbert-base-cased"
145
+ tokenizer = DistilBertTokenizerFast.from_pretrained(model_name,low_cpu_mem_usage=True)
146
 
147
  model_path1 = "DistilBertforTokenclassification"
148