16march_4

app.py

@@ -18,9 +18,9 @@ from torchvision.models import vgg16
 from torchmetrics.image.fid import FrechetInceptionDistance
 from torchmetrics.functional import structural_similarity_index_measure
 from facenet_pytorch import InceptionResnetV1
-from taming.models.vqgan import VQModel
+from taming_modified_new.taming.models.vqgan import VQModel
 from omegaconf import OmegaConf
-from taming.models.vqgan import GumbelVQ
+from taming_modified_new.taming.models.vqgan import GumbelVQ
 import gradio as gr
 from finetunedvqgan import Generator
 from modelz import DeepfakeToSourceTransformer