fresh start

app.py

@@ -478,7 +478,7 @@ def main():
                         outputs=[stats_status, stats_dataframe, stats_status]
                     )
 
-            def submit_request(input_dataset_name, input_split, input_dataset_config, output_dataset_name, prompt_col, model_name, model_rev, sys_prompt,
+            def submit_request(input_dataset_name, input_split, input_dataset_config, output_dataset_name, prompt_col, model_name, sys_prompt,
                                 max_tok, temp, top_k_val, top_p_val, email_addr, num_output_samples, oauth_token=None):
 
                 MASTER_ORG = "synthetic-data-universe/"
@@ -497,7 +497,7 @@ def main():
                         output_dataset_name=MASTER_ORG + output_dataset_name,
                         prompt_column=prompt_col,
                         model_name_or_path=model_name,
-                        model_revision=model_rev,
+                        model_revision="main",
                         model_token=model_token,
                         system_prompt=sys_prompt if sys_prompt else None,
                         max_tokens=int(max_tok),
@@ -522,7 +522,7 @@ def main():
             submit_btn.click(
                 submit_request,
                 inputs=[input_dataset_name, input_dataset_split, input_dataset_config, output_dataset_name, prompt_column, model_name_or_path,
-                        model_revision, system_prompt, max_tokens, temperature, top_k, top_p, email, num_output_samples, current_oauth_token],
+                         system_prompt, max_tokens, temperature, top_k, top_p, email, num_output_samples, current_oauth_token],
                 outputs=output_status
             )