import numpy as np
import torch
import torch.nn as nn
import gradio as gr
from PIL import Music
import torchvision.transforms as transforms