Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -101,10 +101,12 @@ with gr.Blocks(theme="ParityError/Interstellar@0.0.1") as demo:
|
|
| 101 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 102 |
|
| 103 |
# Immediately load models
|
|
|
|
| 104 |
get_tinystyle_model(device)
|
| 105 |
get_style_embedding_model(device)
|
| 106 |
get_luar_model(device)
|
| 107 |
get_mis_model(device)
|
|
|
|
| 108 |
|
| 109 |
gr.Markdown("# TinyStyler Demo")
|
| 110 |
gr.Markdown(
|
|
|
|
| 101 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 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")
|
| 112 |
gr.Markdown(
|