asutosh09 commited on
Commit
91501be
·
verified ·
1 Parent(s): a8c439c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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,15 @@ 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",
27
  description=(
28
+ "ASUTOSH AND BISMAYTOSH"
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
  )
36
 
37
  demo.launch()