# 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)