Update app.py

app.py

@@ -13,7 +13,7 @@ class InferenceWrapper:
         else:
             print(f"Model '{model}' already loaded, reusing it.")
         return self.pipe([seed1, seed2, seed3], w_frames=w_frames)
-wrapper = InferenceWrapper('wikiart-1024')
+wrapper = InferenceWrapper('stylegan3-t-afhqv2-512x512.pkl')
 def fn(s1, s2, s3, w_frames, model):
     return wrapper(s1, s2, s3, w_frames, model)
 gr.Interface(
@@ -23,7 +23,7 @@ gr.Interface(
         gr.inputs.Slider(minimum=0, maximum=999999999, step=1, default=0, label='Random Seed For Image 2'),
         gr.inputs.Slider(minimum=0, maximum=999999999, step=1, default=0, label='Random Seed For Image 3'),
         gr.inputs.Radio([60, 120, 240], type="value", default=60, label='Frames'),
-        gr.inputs.Radio(list(model_map), type="value", default='wikiart-1024', label='Pretrained Model')
+        gr.inputs.Radio(list(model_map), type="value", default='stylegan3-t-afhqv2-512x512.pkl', label='Pretrained Model')
     ],
     outputs='video',
     examples=[[0, 1, 2, 60, 'landscapes-256']],