Update app.py

app.py

@@ -14,7 +14,7 @@ from torch import nn, autograd, optim
 from torch.nn import functional as F
 from tqdm import tqdm
 import lpips
-from model import *
+
 
 #from e4e_projection import projection as e4e_projection
 
@@ -31,6 +31,7 @@ from argparse import Namespace
 from e4e.utils.common import tensor2im
 from e4e.models.psp import pSp
 from e4e.models.encoders import psp_encoders
+from e4e.models.stylegan2.model import Generator
 from util import *
 from huggingface_hub import hf_hub_download