Spaces:

Seetha
/

IMA-pipeline-streamlit

Sleeping

Seetha commited on Oct 20, 2023

Commit

3c8b9ba

1 Parent(s): ccad741

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,7 +142,7 @@ def main():
   st.write("--- %s seconds ---" % (time.time() - start_time))
   model_name = "distilbert-base-cased"
-  tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
   model_path1 = "DistilBertforTokenclassification"

   st.write("--- %s seconds ---" % (time.time() - start_time))
   model_name = "distilbert-base-cased"
+  tokenizer = DistilBertTokenizerFast.from_pretrained(model_name,low_cpu_mem_usage=True)
   model_path1 = "DistilBertforTokenclassification"