Wataru commited on
Commit
2ade89d
·
verified ·
1 Parent(s): dbb3e47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from types import SimpleNamespace
4
  import gradio as gr
5
  import pandas as pd
6
 
7
- # import spaces
8
  import torch
9
 
10
  from utmosv2.utils import get_dataset, get_model
@@ -27,7 +27,7 @@ cfg.phase = "inference"
27
  cfg.weight = None
28
  cfg.num_workers = 1
29
 
30
- # @spaces.GPU
31
  def predict_mos(audio_path: str, domain: str) -> float:
32
  data = pd.DataFrame({"file_path": [audio_path]})
33
  data["dataset"] = domain
 
4
  import gradio as gr
5
  import pandas as pd
6
 
7
+ import spaces
8
  import torch
9
 
10
  from utmosv2.utils import get_dataset, get_model
 
27
  cfg.weight = None
28
  cfg.num_workers = 1
29
 
30
+ @spaces.GPU
31
  def predict_mos(audio_path: str, domain: str) -> float:
32
  data = pd.DataFrame({"file_path": [audio_path]})
33
  data["dataset"] = domain