0Tick commited on
Commit
2d06509
1 Parent(s): 5ac3442

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -25,7 +25,7 @@ class Model:
25
  model = None
26
  tokenizer = None
27
 
28
- available_models = []
29
  current = Model()
30
  job_count = 1
31
 
@@ -37,9 +37,6 @@ def device():
37
  return devices.cpu
38
 
39
 
40
- def list_available_models():
41
- available_models = ["0Tick/e621TagAutocomplete","0Tick/danbooruTagAutocomplete"]
42
-
43
 
44
  def get_model_path(name):
45
  dirname = os.path.join(models_dir, name)
@@ -174,11 +171,11 @@ with gr.Blocks(analytics_enabled=False) as space:
174
  outputs=[res, res_info]
175
  )
176
 
177
- model_selection.change(
178
  fn=model_selection_changed,
179
  inputs=[model_selection],
180
  outputs=[],
181
- )
182
 
183
 
184
  space.launch()
 
25
  model = None
26
  tokenizer = None
27
 
28
+ available_models = ["0Tick/e621TagAutocomplete","0Tick/danbooruTagAutocomplete"]
29
  current = Model()
30
  job_count = 1
31
 
 
37
  return devices.cpu
38
 
39
 
 
 
 
40
 
41
  def get_model_path(name):
42
  dirname = os.path.join(models_dir, name)
 
171
  outputs=[res, res_info]
172
  )
173
 
174
+ model_selection.change(
175
  fn=model_selection_changed,
176
  inputs=[model_selection],
177
  outputs=[],
178
+ )
179
 
180
 
181
  space.launch()