Update app.py

app.py

@@ -13,9 +13,9 @@ import numpy as np
 import torch
 import torch.nn.functional as F
 from collections import OrderedDict
-from AutoPST.onmt_modules.misc import sequence_mask
-from AutoPST.model_autopst import Generator_2 as Predictor
-from AutoPST.hparams_autopst import hparams
+from onmt_modules.misc import sequence_mask
+from model_autopst import Generator_2 as Predictor
+from hparams_autopst import hparams
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -64,8 +64,8 @@ import torch
 import librosa
 import pickle
 import os
-from AutoPST.synthesis import build_model
-from AutoPST.synthesis import wavegen
+from synthesis import build_model
+from synthesis import wavegen
 
 model = build_model().to(device)
 checkpoint = torch.load(hf_hub_download(repo_id="jonathanjordan21/AutoPST", filename="checkpoint_step001000000_ema.pth"), map_location=torch.device('cpu'))