liz/Meatgrinder
liz/Meatgrinder
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Project Alpha 3

Browse Source
This commit is contained in:
Elizabeth Cray 2024-10-10 11:03:15 -04:00
parent 1650f3fccb
commit d3e07ede69
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
summarize.py
View File

@ -1,6 +1,12 @@
from transformers import pipeline from transformers import pipeline
def process(input_text): def process(input_text):
summarizer = pipeline("summarization", model="marianna13/flan-t5-base-summarization", max_length=200, min_length=10, device="mps") summarizer = pipeline(
"summarization",
model="marianna13/flan-t5-base-summarization",
max_length=200,
min_length=10,
device="mps"
)
output = summarizer(input_text)[0]['summary_text'] output = summarizer(input_text)[0]['summary_text']
return output return output

1
transcribe.py
View File

@ -15,6 +15,7 @@ def process(audio_path):
tokenizer=processor.tokenizer, tokenizer=processor.tokenizer,
feature_extractor=processor.feature_extractor, feature_extractor=processor.feature_extractor,
torch_dtype=torch.float32, torch_dtype=torch.float32,
return_timestamps=True,
device="mps" device="mps"
) )
output = pipe(audio_path)["text"] output = pipe(audio_path)["text"]