nateraw commited on
Commit
1fe892b
·
1 Parent(s): 4c42dda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ class InferenceWrapper:
14
  print(f"Model '{model}' already loaded, reusing it.")
15
  return self.pipe([seed1, seed2, seed3], w_frames=w_frames)
16
  wrapper = InferenceWrapper('wikiart-1024')
17
- def fn(seed, model):
18
- return wrapper(seed, model)
19
  gr.Interface(
20
  fn,
21
  inputs=[
 
14
  print(f"Model '{model}' already loaded, reusing it.")
15
  return self.pipe([seed1, seed2, seed3], w_frames=w_frames)
16
  wrapper = InferenceWrapper('wikiart-1024')
17
+ def fn(s1, s2, s3, w_frames, model):
18
+ return wrapper(s1, s2, s3, w_frames, model)
19
  gr.Interface(
20
  fn,
21
  inputs=[