Update app.py

app.py

@@ -3,8 +3,28 @@ from dataset import TranscriptDataset
 from downloader import WhisperPP, YoutubeDownloader
 from interpreter import WhisperInterpreter
 
-def greet(name):
+model_size = "base"
+mode = "transcribe"
+write = False
+download_path = "tmp/"
+
+def dataset(url, name):    
+    ds = TranscriptDataset(name)
+    params = dict(model_size=model_size, language=language, write=write, number_videos=500)
+    data = []
+    whisper_options = dict(
+        model_size=model_size, mode=mode, language=language, write=write
+    )
+    whisperPP = WhisperPP(data, **whisper_options)
+    downloader = YoutubeDownloader(download_path)
+    downloader.download(url, whisperPP)
+    
     return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+yt_input = gr.Textbox(label = 'Youtube Link')
+name_input = gr.Textbox(label = 'Dataset Name')
+
+repo_output = gr.Textbox(label = "Outcome")
+
+iface = gr.Interface(fn=dataset, inputs=[yt_input, name_input], outputs=repo_output)
 iface.launch()