Spaces:
Build error
Build error
File size: 674 Bytes
57341a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# 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)
|