modify model path

app.py

@@ -42,7 +42,7 @@
 import gradio as gr
 import numpy as np
 import time
-from data import write_image_tensor, PatchDataModule, prepare_data, image2tensor, tensor2image
+from data import PatchDataModule, prepare_data, image2tensor, tensor2image
 import torch
 from tqdm import tqdm
 from bigdl.nano.pytorch.trainer import Trainer
@@ -55,7 +55,7 @@ import huggingface_hub
 
 device = 'cpu' 
 dtype = torch.float32
-MODEL_REPO = 'CVPR/FSPBT'
+MODEL_REPO = 'BigDL/FSPBT'
 ckpt_path = huggingface_hub.hf_hub_download(
             MODEL_REPO, 'generator.pt')
 generator = torch.load(ckpt_path)