Wataru commited on
Commit
bba8561
·
verified ·
1 Parent(s): 5a43dca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -28,6 +28,7 @@ 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
 
28
  cfg.num_workers = 1
29
 
30
  @spaces.GPU
31
+ @torch.inference_mode()
32
  def predict_mos(audio_path: str, domain: str) -> float:
33
  data = pd.DataFrame({"file_path": [audio_path]})
34
  data["dataset"] = domain