Update app.py

app.py

@@ -1,3 +1,29 @@
+# import gradio as gr
+
+# gr.load("models/TruongScotl/stvietnamese2").launch()
+
+from transformers import pipeline
 import gradio as gr
 
-gr.load("models/TruongScotl/stvietnamese2").launch()
+model = pipeline("models/TruongScotl/stvietnamese2")
+
+
+def transcribe_audio(mic=None, file=None):
+    if mic is not None:
+        audio = mic
+    elif file is not None:
+        audio = file
+    else:
+        return "You must either provide a mic recording or a file"
+    transcription = model(audio)["text"]
+    return transcription
+
+
+gr.Interface(
+    fn=transcribe_audio,
+    inputs=[
+        gr.Audio(source="microphone", type="filepath", optional=True),
+        gr.Audio(source="upload", type="filepath", optional=True),
+    ],
+    outputs="text",
+).launch()