File size: 608 Bytes
3e533d7
ed6e5d8
 
3e533d7
0d5492e
3e533d7
0d5492e
f9f41df
ed6e5d8
bdec318
 
 
ed6e5d8
 
 
 
f9f41df
 
bdec318
ed6e5d8
bdec318
3e533d7
bada8ef
bdec318
 
 
 
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
from huggingface_hub import login
from diarization import start_diarization
from transcribe import start_transcribe
import ffmpeg
import gradio as gr
import os


def prepare_input(input_file):
    output_file = "input.wav"
    ffmpeg.input(input_file).audio.output(
        output_file, format="wav").run()

    progress = gr.Progress()
    start_diarization(output_file, progress)
    return start_transcribe(progress)


video_interface = gr.Interface(
    fn=prepare_input,
    inputs=gr.Video(type="file"),
    outputs="text",
    title="Test 1"
)

if __name__ == "__main__":
    video_interface.launch()