PRamoneda commited on
Commit
1f1fb78
·
1 Parent(s): 3ad467e
Files changed (2) hide show
  1. app.py +0 -2
  2. get_difficulty.py +9 -1
app.py CHANGED
@@ -117,8 +117,6 @@ demo = gr.Interface(
117
  "Related publication: [IEEE TASLP paper](https://ieeexplore.ieee.org/document/10878288)"
118
  )
119
  )
120
- # activa la cola sin límite de timeout
121
- demo.queue(max_size=None)
122
 
123
  if __name__ == "__main__":
124
  demo.launch(debug=True)
 
117
  "Related publication: [IEEE TASLP paper](https://ieeexplore.ieee.org/document/10878288)"
118
  )
119
  )
 
 
120
 
121
  if __name__ == "__main__":
122
  demo.launch(debug=True)
get_difficulty.py CHANGED
@@ -104,4 +104,12 @@ if __name__ == "__main__":
104
  mp3_path = "yt_audio.mp3"
105
  model_name = "audio_midi_multi_ps_v5"
106
  pred_multi = predict_difficulty(mp3_path, model_name=model_name, rep="multimodal5")
107
- print(f"Predicción dificultad multimodal: {pred_multi}")
 
 
 
 
 
 
 
 
 
104
  mp3_path = "yt_audio.mp3"
105
  model_name = "audio_midi_multi_ps_v5"
106
  pred_multi = predict_difficulty(mp3_path, model_name=model_name, rep="multimodal5")
107
+ print(f"Multimodal: {pred_multi}")
108
+
109
+ model_name = "audio_midi_pianoroll_ps_5_v4"
110
+ pred_multi = predict_difficulty(mp3_path, model_name=model_name, rep="pianoroll5")
111
+ print(f"Pianoroll: {pred_multi}")
112
+
113
+ model_name = "audio_midi_multi_ps_v5"
114
+ pred_multi = predict_difficulty(mp3_path, model_name=model_name, rep="pianoroll5")
115
+ print(f"CQT: {pred_multi}")