Update app.py

app.py

@@ -51,14 +51,18 @@ def run(
     new_dataset_id,
     # progress=gr.Progress(),
 ):
-    ds = load_hf_dataset(ds_name, ds_config, column_name, ds_split)
+    print("Loading dataset")
+    ds = load_hf_dataset(ds_name, ds_config, ds_split)
 
+    print("dataset loaded")
     opt_level = desc2opt[opt_desc]
 
     model_name = model_choice.split()[0]
 
+    print("Loading model/tokenizer")
     model, tokenizer = get_model_and_tokenizer(model_name, opt_level)
 
+    print("Model optimized and loaded")
     batch_embed(
         ds,
         model,