Update app.py

app.py

@@ -17,14 +17,14 @@ tinystyler_module.__spec__.loader.exec_module(tinystyler_module)
     get_tinystyle_model,
     get_style_embedding_model,
     get_luar_model,
-    get_mis_model,
+    get_simcse_model,
     compute_simcse,
     run_tinystyler_batch,
 ) = (
     tinystyler_module.get_tinystyle_model,
     tinystyler_module.get_style_embedding_model,
     tinystyler_module.get_luar_model,
-    tinystyler_module.get_mis_model,
+    tinystyler_module.get_simcse_model,
     tinystyler_module.compute_simcse,
     tinystyler_module.run_tinystyler_batch,
 )
@@ -55,6 +55,7 @@ def run_tinystyler(source_text, target_texts, reranking, temperature, top_p):
             device=device,
             seed=42,
             log_timing=True,
+            sim_func=compute_simcse,
             sim_sample=1,
         )[0]
     except Exception as e:
@@ -110,7 +111,7 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
     get_tinystyle_model(device)
     get_style_embedding_model(device)
     get_luar_model(device)
-    get_mis_model(device)
+    get_simcse_model(device)
     print("Done loading models.", time())
 
     gr.Markdown("# TinyStyler Demo")