Drbrain0620 commited on
Commit
4128dbb
·
verified ·
1 Parent(s): 31d1e52

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
2
  import base64
 
3
  from transformers import pipeline
4
 
5
  # Whisper 모델을 pipeline으로 불러오기
6
- whisper = pipeline("automatic-speech-recognition", model="openai/whisper-base")
7
 
8
  # 음성을 텍스트로 변환하는 함수
9
  def transcribe_audio(audio):
 
1
  import gradio as gr
2
  import base64
3
+ import torch
4
  from transformers import pipeline
5
 
6
  # Whisper 모델을 pipeline으로 불러오기
7
+ whisper = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
8
 
9
  # 음성을 텍스트로 변환하는 함수
10
  def transcribe_audio(audio):