Update app.py

app.py

@@ -29,8 +29,8 @@ def download_from_youtube(youtube_link: str) -> str:
     return "audio.wav"
 
 
-#MODEL_NAME = 'bayartsogt/whisper-large-v2-mn-13'
-MODEL_NAME = 'Dorjzodovsuren/whisper-large-v3-turbo-mn-2'
+MODEL_NAME = 'bayartsogt/whisper-large-v2-mn-13'
+#MODEL_NAME = 'Dorjzodovsuren/whisper-large-v3-turbo-mn-2'
 lang = 'mn'
 
 chunk_length_s = 9
@@ -47,7 +47,7 @@ from transformers import pipeline
 from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
 
 if MODEL_NAME == 'bayartsogt/whisper-large-v2-mn-13':
-  processor = AutoProcessor.from_pretrained('bayartsogt/whisper-large-v2-mn-13')
+  processor = AutoProcessor.from_pretrained(MODEL_NAME)
 
 else:
   processor = AutoProcessor.from_pretrained("openai/whisper-large-v3-turbo")
@@ -132,4 +132,34 @@ def generator(youtube_link, microphone, file_upload, num_speakers, max_duration,
     with open(file_name, 'w') as fp:
         fp.write(history)
     
-    yield history, history, file_name
+    yield history, history, file_name
+
+
+demo = gr.Interface(
+    generator, 
+    inputs=[
+        gr.Textbox(lines=1, placeholder="Paste the URL to a YouTube video here", label="YouTube URL"),
+        gr.Audio(type="filepath"),
+        gr.Audio(type="filepath"),
+        gr.Number(value=1, label="Number of Speakers"),
+        gr.Number(value=120, label="Maximum Duration (Seconds)"),
+        'state',
+    ],
+    outputs=['text', 'state', 'file'],
+    theme="huggingface",
+    title="Transcribe Mongolian Whisper 🇲🇳",
+    description=(
+        "Transcribe Youtube Video / Microphone / Uploaded File in Mongolian Whisper Model." + \
+        " | You can upload SubRip file (`.srt`) [to your youtube video](https://support.google.com/youtube/answer/2734698?hl=en#zippy=%2Cbasic-file-formats)." + \
+        " | Please REFRESH 🔄 the page after you transcribed!" + \
+        " | 🐦 [@_tsogoo_](https://twitter.com/_tsogoo_)" + \
+        " | 🤗 [@bayartsogt](https://huggingface.co/bayartsogt)" + \
+        ""
+    ),
+    allow_flagging="never",
+)
+
+# define queue - required for generators
+demo.queue()
+
+demo.launch(debug=True)