Yoni232 commited on
Commit
e77f67a
·
1 Parent(s): c24f20a

added spaces to requirements and also added gpu decorators in app.py

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +2 -0
  3. requirements.txt +1 -0
README.md CHANGED
@@ -2,7 +2,7 @@
2
  title: Count The Notes
3
  emoji: 🎵
4
  colorFrom: pink
5
- colorTo: green
6
  sdk: gradio
7
  sdk_version: 6.0.2
8
  app_file: app.py
 
2
  title: Count The Notes
3
  emoji: 🎵
4
  colorFrom: pink
5
+ colorTo: blue
6
  sdk: gradio
7
  sdk_version: 6.0.2
8
  app_file: app.py
app.py CHANGED
@@ -6,6 +6,7 @@ using pre-trained models from Hugging Face Hub.
6
  """
7
 
8
  import gradio as gr
 
9
  import tempfile
10
  import os
11
  from pathlib import Path
@@ -35,6 +36,7 @@ def load_model(model_name: str) -> CountEMModel:
35
  return model_cache[model_name]
36
 
37
 
 
38
  def transcribe_audio(
39
  audio_input,
40
  model_choice: str,
 
6
  """
7
 
8
  import gradio as gr
9
+ import spaces
10
  import tempfile
11
  import os
12
  from pathlib import Path
 
36
  return model_cache[model_name]
37
 
38
 
39
+ @spaces.GPU
40
  def transcribe_audio(
41
  audio_input,
42
  model_choice: str,
requirements.txt CHANGED
@@ -15,3 +15,4 @@ wandb>=0.15.0
15
  huggingface_hub>=0.20.0
16
  safetensors>=0.4.3
17
  gradio>=4.0.0
 
 
15
  huggingface_hub>=0.20.0
16
  safetensors>=0.4.3
17
  gradio>=4.0.0
18
+ spaces