asdasd

app.py

@@ -3,10 +3,10 @@ from huggingface_hub import PyTorchModelHubMixin
 import torch
 import matplotlib.pyplot as plt
 import torchvision
-from networks_fastgan import Generator
+from networks_fastgan import MyGenerator
 
 def image_generation(model, number_of_images=1):
-    G = Generator.from_pretrained("cropinky/projected_gan_impressionism")
+    G = MyGenerator.from_pretrained("Cropinky/projected_gan_impressionism")
     return f"generating {number_of_images} images from {model}"
 if __name__ == "__main__":
 

networks_fastgan.py

@@ -147,7 +147,7 @@ class FastganSynthesisCond(nn.Module):
         return self.to_big(feat_last)
 
 
-class Generator(nn.Module, PyTorchModelHubMixin):
+class MyGenerator(nn.Module, PyTorchModelHubMixin):
     def __init__(
         self,
         z_dim=256,