AjayP13 commited on
Commit
0fbe474
·
verified ·
1 Parent(s): c19b76d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,14 +14,14 @@ tinystyler_module = importlib.util.module_from_spec(
14
  )
15
  tinystyler_module.__spec__.loader.exec_module(tinystyler_module)
16
  (
17
- get_tinystyle_model,
18
  get_style_embedding_model,
19
  get_luar_model,
20
  get_simcse_model,
21
  compute_simcse,
22
  run_tinystyler_batch,
23
  ) = (
24
- tinystyler_module.get_tinystyle_model,
25
  tinystyler_module.get_style_embedding_model,
26
  tinystyler_module.get_luar_model,
27
  tinystyler_module.get_simcse_model,
@@ -107,7 +107,7 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
107
 
108
  # Immediately load models
109
  print("Loading models...", time())
110
- get_tinystyle_model(device)
111
  get_style_embedding_model(device)
112
  get_luar_model(device)
113
  get_simcse_model(device)
 
14
  )
15
  tinystyler_module.__spec__.loader.exec_module(tinystyler_module)
16
  (
17
+ get_tinystyler_model,
18
  get_style_embedding_model,
19
  get_luar_model,
20
  get_simcse_model,
21
  compute_simcse,
22
  run_tinystyler_batch,
23
  ) = (
24
+ tinystyler_module.get_tinystyler_model,
25
  tinystyler_module.get_style_embedding_model,
26
  tinystyler_module.get_luar_model,
27
  tinystyler_module.get_simcse_model,
 
107
 
108
  # Immediately load models
109
  print("Loading models...", time())
110
+ get_tinystyler_model(device)
111
  get_style_embedding_model(device)
112
  get_luar_model(device)
113
  get_simcse_model(device)