Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import pandas as pd 
 from realtabformer import REaLTabFormer
+from scipy.io import arff
 
 rtf_model = REaLTabFormer(
     model_type="tabular",
@@ -8,7 +9,8 @@ rtf_model = REaLTabFormer(
 
 
 def generate_data(file):
-    df = pd.read_csv(file)
+    data = arff.loadarff(file)
+    df = pd.Dataframe(data[0])
     rtf_model.fit(df)
     # Generate synthetic data
     samples = rtf_model.sample(n_samples=5)
@@ -111,7 +113,7 @@ with gr.Blocks(css = css) as demo:
         #     transcribe_audio_r = gr.Button('Transcribe')
         
         with gr.Tab("Upload Data as File"):
-            data_input_u = gr.File(label = 'Upload Data File', file_types=["text", ".json", ".csv"])
+            data_input_u = gr.File(label = 'Upload Data File', file_types=["text", ".json", ".csv", ".arff"])
             generate_data_btn = gr.Button('Generate Synthetic Data')
 
         with gr.Row():