Spaces:

tinystyler
/

tinystyler_demo

Running

zacharyhorv commited on Nov 14, 2024

Commit

d32e977

verified ·

1 Parent(s): 82e50f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
     # Immediately load models
     print("Loading models...", time())
-    get_tinystyler_model(device)
     get_style_embedding_model(device)
     get_luar_model(device)
     get_simcse_model(device)

     # Immediately load models
     print("Loading models...", time())
+    get_tinystyler_model(device, model_name='tinystyler_sim')
     get_style_embedding_model(device)
     get_luar_model(device)
     get_simcse_model(device)