Spaces:

Ho-ot
/

Learnlite

Sleeping

Ho-ot commited on May 28, 2024

Commit

75aba89

verified ·

1 Parent(s): ebd2a36

Update ocr.py

Files changed (1) hide show

ocr.py CHANGED Viewed

@@ -15,7 +15,7 @@ from firebase_admin import firestore
 @st.cache_resource
 def load_model():
-    model_path = "facebook/bart-large-cnn"
     tokenizer = AutoTokenizer.from_pretrained(model_path,use_fast=False)
     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