add Umamusume model

README.md

@@ -4,7 +4,7 @@ emoji: 😊🎙️
 colorFrom: red
 colorTo: pink
 sdk: gradio
-sdk_version: 3.12.0
+sdk_version: 3.16.1
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -3,6 +3,7 @@ import json
 import os
 import re
 import tempfile
+from pathlib import Path
 
 import librosa
 import numpy as np
@@ -19,23 +20,14 @@ from mel_processing import spectrogram_torch
 
 limitation = os.getenv("SYSTEM") == "spaces"  # limit text and audio length in huggingface spaces
 
+audio_postprocess_ori = gr.Audio.postprocess
+
 
 def audio_postprocess(self, y):
-    if y is None:
+    data = audio_postprocess_ori(self, y)
+    if data is None:
         return None
-
-    if gr_utils.validate_url(y):
-        file = gr_processing_utils.download_to_file(y, dir=self.temp_dir)
-    elif isinstance(y, tuple):
-        sample_rate, data = y
-        file = tempfile.NamedTemporaryFile(
-            suffix=".wav", dir=self.temp_dir, delete=False
-        )
-        gr_processing_utils.audio_to_file(sample_rate, data, file.name)
-    else:
-        file = gr_processing_utils.create_tmp_copy_of_file(y, dir=self.temp_dir)
-
-    return gr_processing_utils.encode_url_or_file_to_base64(file.name)
+    return gr_processing_utils.encode_url_or_file_to_base64(data["name"])
 
 
 gr.Audio.postprocess = audio_postprocess

export_model.py

@@ -1,8 +1,8 @@
 import torch
 
 if __name__ == '__main__':
-    model_path = "saved_model/11/model.pth"
-    output_path = "saved_model/11/model1.pth"
+    model_path = "saved_model/18/model.pth"
+    output_path = "saved_model/18/model1.pth"
     checkpoint_dict = torch.load(model_path, map_location='cpu')
     checkpoint_dict_new = {}
     for k, v in checkpoint_dict.items():

saved_model/18/config.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aaa921889da820eee740426dc28f8b11242d5ea64635d75460b04d0f4c830f83
+size 3336

saved_model/18/model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a53f4eb6bf8226b3fb4a3b31436235f697692f5566039ce3491b80af9a9567a
+size 158962765

saved_model/info.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8b71ff5dab7d60ce8fdbe75fa48b607574b845b4b57fa119299e1f21216d41e7
-size 4697
+oid sha256:b6c673fd78f290cc81ee399069e2bd68125c768658f0a46e320070255cd93bab
+size 4889