Spaces:

Ho-ot
/

Learnlite

Sleeping

Ho-ot commited on May 11, 2024

Commit

11a48fe

verified ·

1 Parent(s): 1c97482

Update ocr.py

Files changed (1) hide show

ocr.py CHANGED Viewed

@@ -15,7 +15,7 @@ import textwrap
 @st.cache_resource
 def load_model():
     model_path = "Ho-ot/pegasus-fyp"
-    tokenizer = AutoTokenizer.from_pretrained(model_path)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
     return tokenizer, model

 @st.cache_resource
 def load_model():
     model_path = "Ho-ot/pegasus-fyp"
+    tokenizer = AutoTokenizer.from_pretrained(model_path,use_fast=False)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
     return tokenizer, model