Hugging Face's logo

Spaces:
saharM
/
dialogue_summarizer
Sleeping

App Files Community
saharM commited on
Commit
21ed461
·
verified ·
1 Parent(s): 49f382c

fix model loading

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -68,9 +68,10 @@ st.markdown("""
68
 
69
  # Load model and tokenizer
70
  MODEL_PATH = "./models/fine-tuned_bart_base"
71
- model = BartForConditionalGeneration.from_pretrained(MODEL_PATH)
72
  model = model. cpu()
73
- tokenizer = BartTokenizer.from_pretrained(MODEL_PATH)
 
74
 
75
  #Helper functions
76
  def extract_speakers(dialogue):
 
68
 
69
  # Load model and tokenizer
70
  MODEL_PATH = "./models/fine-tuned_bart_base"
71
+ model = BartForConditionalGeneration.from_pretrained(MODEL_PATH, local_files_only=True)
72
  model = model. cpu()
73
+ tokenizer = BartTokenizer.from_pretrained(MODEL_PATH, local_files_only=True)
74
+
75
 
76
  #Helper functions
77
  def extract_speakers(dialogue):