Spaces:
Runtime error
Runtime error
import whisper | |
import pyaudio | |
import tkinter as tk | |
# PyAudio config | |
CHUNK = 1024 | |
FORMAT = pyaudio.paInt16 | |
CHANNELS = 1 | |
RATE = 44100 | |
# Create GUI | |
root = tk.Tk() | |
text_box = tk.Text(root) | |
text_box.pack() | |
# Load Whisper | |
model = whisper.load_model("base") | |
# Start audio stream | |
p = pyaudio.PyAudio() | |
stream = p.open(format=FORMAT, | |
channels=CHANNELS, | |
rate=RATE, | |
input=True, | |
frames_per_buffer=CHUNK) | |
# Listen to mic in a loop | |
while True: | |
data = stream.read(CHUNK) | |
text = model.transcribe(data)['text'] | |
# Insert into GUI | |
text_box.insert('end', text) | |
text_box.see('end') | |
root.update() | |
stream.stop_stream() | |
stream.close() | |
p.terminate() | |