Spaces:

nateraw
/

stylegan3

Runtime error

nateraw commited on Apr 9, 2022

Commit

3aa7212

•

1 Parent(s): c119503

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ class InferenceWrapper:
             print(f"Model '{model}' already loaded, reusing it.")
         return self.pipe(seed)
-wrapper = InferenceWrapper('wikiart-1024')
 def fn(seed, model):
     return wrapper(seed, model)

             print(f"Model '{model}' already loaded, reusing it.")
         return self.pipe(seed)
+wrapper = InferenceWrapper('stylegan3-t-afhqv2-512x512.pkl')
 def fn(seed, model):
     return wrapper(seed, model)