my_first_draft / drafts_1.py
ArseniyPerchik's picture
more
57341a2
raw
history blame contribute delete
674 Bytes
# from transformers import pipeline
# import gradio as gr
#
# pipe = pipeline("text-classification")
#
#
# # Access the model and tokenizer
# model = pipe.model
# tokenizer = pipe.tokenizer
#
# # Save to a local directory
# save_directory = "./tc"
# model.save_pretrained(save_directory)
# tokenizer.save_pretrained(save_directory)
# ---
from transformers import pipeline
import gradio as gr
pipe = pipeline("automatic-speech-recognition", "facebook/wav2vec2-base-960h")
# Save to a local directory
save_directory = "./asr"
pipe.model.save_pretrained(save_directory)
pipe.feature_extractor.save_pretrained(save_directory)
pipe.tokenizer.save_pretrained(save_directory)