Update app.py

app.py

@@ -8,7 +8,7 @@ rtf_model = REaLTabFormer(
     gradient_accumulation_steps=1)
 
 
-def generate_data(file):
+def generate_data(file, num_samples):
     if '.arff' in file.name:
         data = arff.loadarff(open(file.name,'rt'))
         df = pd.DataFrame(data[0])
@@ -16,7 +16,7 @@ def generate_data(file):
         df = pd.read_csv(file.name)
     rtf_model.fit(df)
     # Generate synthetic data
-    samples = rtf_model.sample(n_samples=5)
+    samples = rtf_model.sample(n_samples=num_samples)
 
     return df.head(),samples