ACloudCenter commited on
Commit
145285e
·
1 Parent(s): cbb1a17

Add gpu decorator

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -16,6 +16,7 @@ model = MoonshineForConditionalGeneration.from_pretrained('UsefulSensors/moonshi
16
  processor = AutoProcessor.from_pretrained('UsefulSensors/moonshine-tiny')
17
 
18
  # Function to transcribe audio
 
19
  def transcribe_audio(audio_file):
20
  # Load audio file
21
  dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
 
16
  processor = AutoProcessor.from_pretrained('UsefulSensors/moonshine-tiny')
17
 
18
  # Function to transcribe audio
19
+ @spaces.GPU
20
  def transcribe_audio(audio_file):
21
  # Load audio file
22
  dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")