Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|