Lisandro commited on
Commit
7f5d78f
·
1 Parent(s): 4daeaec

feat: Fix selected space loading in app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -48,6 +48,8 @@ def infer(selected_space_index, prompt, seed=42, randomize_seed=False, width=102
48
  )
49
  result = job.result()
50
  except ValueError as e:
 
 
51
  raise gr.Error(e)
52
 
53
  return (selected_space_index, ) + result
 
48
  )
49
  result = job.result()
50
  except ValueError as e:
51
+ client = None
52
+ selected_space_index = (selected_space_index + 1) % len(flux_1_schell_spaces)
53
  raise gr.Error(e)
54
 
55
  return (selected_space_index, ) + result