utils/transcriber.py

from gradio_client import Client, handle_file
from utils.api_configs import api_configs

def transcriber(invideo_file:str, 
                max_words_per_line:int,
                task:str,
                model_version:str,
                config_file:str
                ):
        HF_TOKEN = api_configs(config_file)["secrets"]["hf-token"]
        HF_SPACE = api_configs(config_file)["secrets"]["hf-space"]
        client = Client(HF_SPACE, hf_token=HF_TOKEN)
        result = client.predict(
                file_input=handle_file(invideo_file),
                file_type = "video",
                max_words_per_line=max_words_per_line,
                task=task,
                model_version=model_version,
                api_name="/predict"
        )
        return result[0]