multimodalart HF staff commited on
Commit
20c1d49
1 Parent(s): e665eb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -80,9 +80,9 @@ def process_image(
80
  width=width,
81
  height=height,
82
  generator=torch.Generator().manual_seed(seed)
83
- )
84
 
85
- return result[0], seed
86
 
87
  # UI CSS
88
  css = """
 
80
  width=width,
81
  height=height,
82
  generator=torch.Generator().manual_seed(seed)
83
+ ).images[0]
84
 
85
+ return result, seed
86
 
87
  # UI CSS
88
  css = """