Upload folder using huggingface_hub

README.md

@@ -1,12 +1,6 @@
 ---
-title: Transcript And Translate
-emoji: 🦀
-colorFrom: blue
-colorTo: blue
+title: transcript_and_translate
+app_file: main.py
 sdk: gradio
 sdk_version: 5.21.0
-app_file: app.py
-pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from transformers import pipeline
+
+
+whisper = pipeline("automatic-speech-recognition", model="openai/whisper-small")
+trans_en_ko = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ko")
+trans_ko_en = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
+def transcribe_and_translate(audio):
+    text = whisper(audio)['text'] #type: ignore
+    trans_text = trans_en_ko(text)['translation_text'] #type: ignore
+
+    return text, trans_text
+
+demo = gr.Interface(transcribe_and_translate, 'audio', ['text', 'text'])
+demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+torch, torchaudio, transformers, gradio
+torch
+torchaudio
+gradio
+transformers