Wataru commited on
Commit
95a2b02
1 Parent(s): 5f847a6

updated dependency

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +113 -9
app.py CHANGED
@@ -39,7 +39,7 @@ def predict_mos(audio_path: str, domain: str, quick: bool) -> float:
39
  model = get_model(cfg, device).eval()
40
  for _ in range(5):
41
  test_dataset = get_dataset(cfg, data, "test")
42
- p = model(*[torch.tensor(t).unsqueeze(0).to(device) for t in test_dataset[0][:-1]])
43
  preds += p.cpu().numpy()[0][0]
44
  if quick:
45
  return preds
 
39
  model = get_model(cfg, device).eval()
40
  for _ in range(5):
41
  test_dataset = get_dataset(cfg, data, "test")
42
+ p = model(*[torch.tensor(t,dtype=torch.float32).unsqueeze(0).to(device) for t in test_dataset[0][:-1]])
43
  preds += p.cpu().numpy()[0][0]
44
  if quick:
45
  return preds
requirements.txt CHANGED
@@ -1,10 +1,114 @@
1
- numpy>=1.24.4
2
- pandas>=2.2.2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  torch==2.2.0
4
- timm>=1.0.7
5
- librosa>=0.10.2
6
- tqdm>=4.66.4
7
- scikit-learn>=1.3.2
8
- transformers>=4.42.4
9
- wandb>=0.17.0
10
- python-dotenv>=1.0.1
 
 
 
 
 
 
 
1
+ aiofiles==23.2.1
2
+ annotated-types==0.7.0
3
+ anyio==4.4.0
4
+ audioread==3.0.1
5
+ certifi==2024.7.4
6
+ cffi==1.16.0
7
+ charset-normalizer==3.3.2
8
+ click==8.1.7
9
+ contourpy==1.2.1
10
+ cycler==0.12.1
11
+ decorator==5.1.1
12
+ dnspython==2.6.1
13
+ docker-pycreds==0.4.0
14
+ email_validator==2.2.0
15
+ fastapi==0.111.1
16
+ fastapi-cli==0.0.4
17
+ ffmpy==0.4.0
18
+ filelock==3.15.4
19
+ fonttools==4.53.1
20
+ fsspec==2024.6.1
21
+ gitdb==4.0.11
22
+ GitPython==3.1.43
23
+ gradio==4.40.0
24
+ gradio_client==1.2.0
25
+ h11==0.14.0
26
+ httpcore==1.0.5
27
+ httptools==0.6.1
28
+ httpx==0.27.0
29
+ huggingface-hub==0.24.5
30
+ idna==3.7
31
+ importlib_resources==6.4.0
32
+ Jinja2==3.1.4
33
+ joblib==1.4.2
34
+ kiwisolver==1.4.5
35
+ lazy_loader==0.4
36
+ librosa==0.10.2.post1
37
+ llvmlite==0.43.0
38
+ markdown-it-py==3.0.0
39
+ MarkupSafe==2.1.5
40
+ matplotlib==3.9.1
41
+ mdurl==0.1.2
42
+ mpmath==1.3.0
43
+ msgpack==1.0.8
44
+ networkx==3.3
45
+ numba==0.60.0
46
+ numpy==1.26.4
47
+ nvidia-cublas-cu12==12.1.3.1
48
+ nvidia-cuda-cupti-cu12==12.1.105
49
+ nvidia-cuda-nvrtc-cu12==12.1.105
50
+ nvidia-cuda-runtime-cu12==12.1.105
51
+ nvidia-cudnn-cu12==8.9.2.26
52
+ nvidia-cufft-cu12==11.0.2.54
53
+ nvidia-curand-cu12==10.3.2.106
54
+ nvidia-cusolver-cu12==11.4.5.107
55
+ nvidia-cusparse-cu12==12.1.0.106
56
+ nvidia-nccl-cu12==2.19.3
57
+ nvidia-nvjitlink-cu12==12.5.82
58
+ nvidia-nvtx-cu12==12.1.105
59
+ orjson==3.10.6
60
+ packaging==24.1
61
+ pandas==2.2.2
62
+ pillow==10.4.0
63
+ platformdirs==4.2.2
64
+ pooch==1.8.2
65
+ protobuf==5.27.3
66
+ psutil==5.9.8
67
+ pycparser==2.22
68
+ pydantic==2.8.2
69
+ pydantic_core==2.20.1
70
+ pydub==0.25.1
71
+ Pygments==2.18.0
72
+ pyparsing==3.1.2
73
+ python-dateutil==2.9.0.post0
74
+ python-dotenv==1.0.1
75
+ python-multipart==0.0.9
76
+ pytz==2024.1
77
+ PyYAML==6.0.1
78
+ regex==2024.7.24
79
+ requests==2.32.3
80
+ rich==13.7.1
81
+ ruff==0.5.5
82
+ safetensors==0.4.3
83
+ scikit-learn==1.5.1
84
+ scipy==1.14.0
85
+ semantic-version==2.10.0
86
+ sentry-sdk==2.12.0
87
+ setproctitle==1.3.3
88
+ shellingham==1.5.4
89
+ six==1.16.0
90
+ smmap==5.0.1
91
+ sniffio==1.3.1
92
+ soundfile==0.12.1
93
+ soxr==0.4.0
94
+ spaces==0.29.2
95
+ starlette==0.37.2
96
+ sympy==1.13.1
97
+ threadpoolctl==3.5.0
98
+ timm==1.0.8
99
+ tokenizers==0.19.1
100
+ tomlkit==0.12.0
101
  torch==2.2.0
102
+ torchvision==0.17.0
103
+ tqdm==4.66.4
104
+ transformers==4.43.3
105
+ triton==2.2.0
106
+ typer==0.12.3
107
+ typing_extensions==4.12.2
108
+ tzdata==2024.1
109
+ urllib3==2.2.2
110
+ uvicorn==0.30.4
111
+ uvloop==0.19.0
112
+ wandb==0.17.5
113
+ watchfiles==0.22.0
114
+ websockets==12.0