Solved

app.py

@@ -15,7 +15,7 @@ import nltk
 nltk.download('wordnet')
 
 # @title Load Model
-selected_model = 'landshapes-v1'
+selected_model = 'landshapes-v2'
 
 # Load model
 

models/wrappers.py

@@ -109,7 +109,7 @@ class StyleGAN2(BaseModel):
         # Image widths
         configs = {
             # Converted NVIDIA official
-            'landshapes-v1': 1024,
+            'landshapes-v2': 1024,
             'antarctica': 1024,
             'ffhq': 1024,
             'car': 512,
@@ -144,7 +144,7 @@ class StyleGAN2(BaseModel):
     # URLs created with https://sites.google.com/site/gdocs2direct/
     def download_checkpoint(self, outfile):
         checkpoints = {
-            'landshapes-v1': 'https://drive.google.com/uc?export=download&id=1cvnfOyQPPL6ccWKHmmtBIabeZ7xZyDJ9',
+            'landshapes-v2': 'https://drive.google.com/uc?export=download&id=1c5XOjMqaCs4nQ4pAsC653WdAijPSUd_5',
             'antarctica': 'https://drive.google.com/uc?export=download&id=1IGxfa-TLZDgnPsctUzJ7rYyLiFr4cY2M',
             'horse': 'https://drive.google.com/uc?export=download&id=18SkqWAkgt0fIwDEf2pqeaenNi4OoCo-0',
             'ffhq': 'https://drive.google.com/uc?export=download&id=1FJRwzAkV-XWbxgTwxEmEACvuqF5DsBiV',