RamAnanth1 commited on
Commit
ffd7c99
1 Parent(s): 58e5dd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from scipy.io import arff
5
 
6
  rtf_model = REaLTabFormer(
7
  model_type="tabular",
8
- epochs=50, # Default is 200
9
  gradient_accumulation_steps=4)
10
 
11
 
@@ -128,7 +128,7 @@ with gr.Blocks(css = css) as demo:
128
 
129
 
130
  generate_data_btn.click(generate_data, inputs = [data_input_u,num_samples], outputs = [data_output])
131
- examples = gr.Examples(examples=[['diabetes.arff',5]],inputs = [data_input_u,num_samples], outputs = [data_output], cache_examples = True, fn = generate_data)
132
 
133
 
134
  demo.launch()
 
5
 
6
  rtf_model = REaLTabFormer(
7
  model_type="tabular",
8
+ epochs=25, # Default is 200
9
  gradient_accumulation_steps=4)
10
 
11
 
 
128
 
129
 
130
  generate_data_btn.click(generate_data, inputs = [data_input_u,num_samples], outputs = [data_output])
131
+ examples = gr.Examples(examples=[['diabetes.arff',5], ["titanic.csv", 15]],inputs = [data_input_u,num_samples], outputs = [data_output], cache_examples = True, fn = generate_data)
132
 
133
 
134
  demo.launch()