asutosh09 commited on
Commit
acfcbda
·
verified ·
1 Parent(s): 4344217

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -2,7 +2,7 @@ import torch
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
 
@@ -25,13 +25,11 @@ demo = gr.Interface(
25
  outputs=gr.Label(),
26
  title="Music Genre Classification",
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
  )
36
-
37
  demo.launch()
 
2
  from transformers import pipeline
3
  import gradio as gr
4
 
5
+ MODEL_NAME = "Music Genre classification_NIST"
6
 
7
  device = 0 if torch.cuda.is_available() else "cpu"
8
 
 
25
  outputs=gr.Label(),
26
  title="Music Genre Classification",
27
  description=(
28
+ "Project by-Asutosh Panda and Bismaytosh Malik."
 
 
29
  ),
30
  examples="./examples",
31
  cache_examples=True,
32
  allow_flagging="never",
33
  )
34
+ share=True in launch()
35
  demo.launch()