Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -92,11 +92,7 @@ if torch.cuda.is_available():
|
|
92 |
pipe = pipe.to("cuda")
|
93 |
|
94 |
def get_model_list():
|
95 |
-
|
96 |
-
for model in models:
|
97 |
-
if Path(model[1]).is_dir():
|
98 |
-
model_available.append(model)
|
99 |
-
return model_available
|
100 |
|
101 |
|
102 |
unet_cache = dict()
|
|
|
92 |
pipe = pipe.to("cuda")
|
93 |
|
94 |
def get_model_list():
|
95 |
+
return models
|
|
|
|
|
|
|
|
|
96 |
|
97 |
|
98 |
unet_cache = dict()
|