m7k-run commited on
Commit
0b104f9
·
verified ·
1 Parent(s): d9ff8c9

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
4
 
5
  # Load model directly from HF Hub
6
- repo_name = "m7k-run/ast-genre-classifier"
7
  model = AutoModelForAudioClassification.from_pretrained(repo_name)
8
  feature_extractor = AutoFeatureExtractor.from_pretrained(repo_name)
9
 
@@ -23,7 +23,7 @@ def predict(audio):
23
 
24
  demo = gr.Interface(
25
  fn=predict,
26
- inputs=gr.Audio(type="numpy"),
27
  outputs=gr.Label(),
28
  title="Audio Classifier"
29
  )
 
3
  from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
4
 
5
  # Load model directly from HF Hub
6
+ repo_name = "your-hf-username/your-model-name"
7
  model = AutoModelForAudioClassification.from_pretrained(repo_name)
8
  feature_extractor = AutoFeatureExtractor.from_pretrained(repo_name)
9
 
 
23
 
24
  demo = gr.Interface(
25
  fn=predict,
26
+ inputs=gr.Audio(type="numpy"),
27
  outputs=gr.Label(),
28
  title="Audio Classifier"
29
  )