2 hf

README.md

@@ -9,7 +9,4 @@ app_file: app.py
 pinned: false
 license: mit
 short_description: Piano Sound Quality Classifier
----
-
-# Cite
-[1] <a href="https://arxiv.org/pdf/2310.04722.pdf">Monan Zhou, Shangda Wu, Shaohua Ji, Zijin Li, and Wei Li. A Holistic Evaluation of Piano Sound Quality[C]//Proceedings of the 10th Conference on Sound and Music Technology (CSMT). Springer, Singapore, 2023.</a>
+---

model.py

@@ -1,12 +1,9 @@
 import torch
 import torch.nn as nn
-from modelscope import snapshot_download
+from huggingface_hub import snapshot_download
 from torchvision.models import squeezenet1_1
 
-MODEL_DIR = snapshot_download(
-    "ccmusic-database/pianos",
-    cache_dir="./__pycache__",
-)
+MODEL_DIR = snapshot_download("ccmusic-database/pianos", cache_dir="./__pycache__")
 
 
 def Classifier(cls_num=8, output_size=512, linear_output=False):

requirements.txt

@@ -2,5 +2,4 @@ torch
 pillow
 librosa
 matplotlib
-torchvision
-modelscope[framework]==1.18
+torchvision