Update app.py

app.py

@@ -17,13 +17,15 @@ 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,
 )
 
@@ -53,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,
         )[0]
     except Exception as e:
         return f"Error: {e}"
@@ -107,7 +110,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")