Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import torch
|
|
| 2 |
from transformers import pipeline
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
-
MODEL_NAME = "
|
| 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
|
| 27 |
description=(
|
| 28 |
-
"
|
| 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()
|