Spaces:

sparkleai
/

stt

Paused

komyu1227 commited on Apr 11

Commit

2f0576c

1 Parent(s): 3e496b9

fix code

Files changed (1) hide show

serve.py CHANGED Viewed

@@ -16,7 +16,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # model = load_model(device)
-processor = WhisperProcessor.from_pretrained("Ivydata/whisper-small-japanese").to(device)
 model = WhisperForConditionalGeneration.from_pretrained("Ivydata/whisper-small-japanese").to(device)
 def transcribe_audio(audio_data_bytes):

 # model = load_model(device)
+processor = WhisperProcessor.from_pretrained("Ivydata/whisper-small-japanese")
 model = WhisperForConditionalGeneration.from_pretrained("Ivydata/whisper-small-japanese").to(device)
 def transcribe_audio(audio_data_bytes):