Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from transformers import pipeline
3
  import gradio as gr
4
 
5
- MODEL_NAME = "MJackismyShephard/whisper-medium.en-finetuned-gtzan"
6
 
7
  device = 0 if torch.cuda.is_available() else "cpu"
8
 
@@ -23,16 +23,13 @@ demo = gr.Interface(
23
  fn=classify_audio,
24
  inputs= gr.Audio(label="Audio file", type="filepath"),
25
  outputs=gr.Label(),
26
- title="Music Genre Classification by Asutosh Panda and Bismaytosh Malik",
27
  description=(
28
- "Classify long-form audio or microphone inputs with the click of a button! Demo uses the"
29
- f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to classify audio files"
30
- " of arbitrary length."
31
  ),
32
  examples="./examples",
33
  cache_examples=True,
34
  allow_flagging="never",
35
- share=True
36
  )
37
 
38
  demo.launch()
 
2
  from transformers import pipeline
3
  import gradio as gr
4
 
5
+ MODEL_NAME = "JackismyShephard/whisper-medium.en-finetuned-gtzan"
6
 
7
  device = 0 if torch.cuda.is_available() else "cpu"
8
 
 
23
  fn=classify_audio,
24
  inputs= gr.Audio(label="Audio file", type="filepath"),
25
  outputs=gr.Label(),
26
+ title="Music Genre Classification by ASUTOSH and BISMAYTOSH MALIK",
27
  description=(
28
+ "Nothing to share as description"
 
 
29
  ),
30
  examples="./examples",
31
  cache_examples=True,
32
  allow_flagging="never",
 
33
  )
34
 
35
  demo.launch()