parthb3 commited on
Commit
e9e83c6
·
1 Parent(s): e72d884

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ def get_summary(audio_text):
97
  tokenizer = AutoTokenizer.from_pretrained(llm_model) # set tockenizer
98
 
99
  from transformers import pipeline
100
- summarizer = pipeline("summarization", model=llm_model) # set summarizer
101
 
102
  model_max_tokens = tokenizer.model_max_length # get max tockens model can process
103
  text_tokens = len(tokenizer.tokenize(audio_text)) # get number of tockens in audio text
 
97
  tokenizer = AutoTokenizer.from_pretrained(llm_model) # set tockenizer
98
 
99
  from transformers import pipeline
100
+ summarizer = pipeline("summarization", model=llm_model, from_pt=True) # set summarizer
101
 
102
  model_max_tokens = tokenizer.model_max_length # get max tockens model can process
103
  text_tokens = len(tokenizer.tokenize(audio_text)) # get number of tockens in audio text