nateraw commited on
Commit
3aa7212
1 Parent(s): c119503

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ class InferenceWrapper:
19
  print(f"Model '{model}' already loaded, reusing it.")
20
  return self.pipe(seed)
21
 
22
- wrapper = InferenceWrapper('wikiart-1024')
23
  def fn(seed, model):
24
  return wrapper(seed, model)
25
 
 
19
  print(f"Model '{model}' already loaded, reusing it.")
20
  return self.pipe(seed)
21
 
22
+ wrapper = InferenceWrapper('stylegan3-t-afhqv2-512x512.pkl')
23
  def fn(seed, model):
24
  return wrapper(seed, model)
25