Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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()
|