Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -102,10 +102,10 @@ with gr.Blocks(theme="ParityError/[email protected]") 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")
|