Spaces:
Running
Running
feat: add wav2vec2_model attribute to DemoArtifacts for model integration
Browse files- local_demo.py +1 -0
local_demo.py
CHANGED
|
@@ -83,6 +83,7 @@ class DemoArtifacts:
|
|
| 83 |
feature_importance: dict[str, float]
|
| 84 |
feature_stats: dict[str, Any]
|
| 85 |
dataset_summary: dict[str, Any]
|
|
|
|
| 86 |
|
| 87 |
|
| 88 |
def _safe_model_name(name: str) -> str:
|
|
|
|
| 83 |
feature_importance: dict[str, float]
|
| 84 |
feature_stats: dict[str, Any]
|
| 85 |
dataset_summary: dict[str, Any]
|
| 86 |
+
wav2vec2_model: Any # Wav2Vec2MusicClassifier | None
|
| 87 |
|
| 88 |
|
| 89 |
def _safe_model_name(name: str) -> str:
|