Update app.py

app.py

@@ -14,7 +14,7 @@ from diffusers.utils.import_utils import is_xformers_available
 
 from src.my_utils.testing_utils import parse_args_paired_testing
 from src.de_net import DEResNet
-from s3diff_tile import S3Diff
+from src.s3diff_tile import S3Diff
 from torchvision import transforms
 from utils.wavelet_color import wavelet_color_fix, adain_color_fix
 
@@ -24,9 +24,12 @@ tensor_transforms = transforms.Compose([
 
 args = parse_args_paired_testing()
 
+# Run the script to get pretrained models
+subprocess.run(["bash", "get_pretrained_models.sh"])
+
 # Load scheduler, tokenizer and models.
-pretrained_model_path = 'checkpoint-path/s3diff.pkl'
-t2i_path = 'sd-turbo-path'
+pretrained_model_path = 'checkpoints/s3diff.pkl'
+t2i_path = 'stabilityai/sd-turbo'
 de_net_path = 'assets/mm-realsr/de_net.pth'
 
 # initialize net_sr