Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -102,10 +102,10 @@ with gr.Blocks(theme="ParityError/Interstellar@0.0.1") as demo:
|
|
| 102 |
|
| 103 |
# Immediately load models
|
| 104 |
print("Loading models...")
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
print("Done loading models.")
|
| 110 |
|
| 111 |
gr.Markdown("# TinyStyler Demo")
|
|
|
|
| 102 |
|
| 103 |
# Immediately load models
|
| 104 |
print("Loading models...")
|
| 105 |
+
get_tinystyle_model(device)
|
| 106 |
+
get_style_embedding_model(device)
|
| 107 |
+
get_luar_model(device)
|
| 108 |
+
get_mis_model(device)
|
| 109 |
print("Done loading models.")
|
| 110 |
|
| 111 |
gr.Markdown("# TinyStyler Demo")
|