Update app.py

app.py

@@ -9,18 +9,29 @@ import torchcrepe
 from infer import *
 import logging
 from infer_tools.infer_tool import *
-import gradio as gr
-import transformers
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-# define the input interface for the app
-inputs = gr.inputs.Textbox(lines=1, label="Enter a link:")
-inputs += gr.inputs.Audio(label="Upload a WAV file:")
+##EDIT FOR CPU
+# Open the file and read it into a string
+with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py") as f:
+    text = f.read()
 
-# define the output interface for the app
-outputs = gr.outputs.Textbox(lines=1, label="Diff-SVC prediction:")
+# Replace the original line with the new line
+text = text.replace("def load(f, map_location=None, pickle_module=pickle, **pickle_load_args):", "def load(f, map_location='cpu', pickle_module=pickle, **pickle_load_args):")
+# Save the modified string to the original file
+with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py", "w") as f:
+    f.write(text)
 
-# define the app itself
-app = gr.Interface(model, inputs, outputs, title="DIFF-SVC Inference Cloud")
+print("Replaced")
+with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py") as f:
+    text = f.read()
+print(text)
 
-# run the app
-app.launch()
+############
+logging.getLogger('numba').setLevel(logging.WARNING)
+
+# 工程文件夹名，训练时用的那个
+project_name = "Unnamed"
+model_path = f'./checkpoints/Unnamed/model_ckpt_steps_192000.ckpt'
+config_path=f'./checkpoints/Unnamed/config.yaml'
+hubert_gpu=False
+svc_model = Svc(project_name,config_path,hubert_gpu, model_path)
+print('model loaded')