Spaces:
Runtime error
Runtime error
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()
|