Update app.py

app.py

@@ -1,20 +1,18 @@
+import os
+import sys
 import gradio as gr
 from model.model import fine_tune
 from data.preprocess import load_data, preprocess_data, save_processed_data
 
 def prepare_and_train(model_name, dataset_path, epochs, batch_size, learning_rate):
-    try:
-        # Load and preprocess the dataset
-        data = load_data(dataset_path)
-        cleaned_data = preprocess_data(data)
-        processed_data_path = 'data/processed/processed_dataset.csv'
-        save_processed_data(cleaned_data, processed_data_path)
+    # Load and preprocess the dataset
+    data = load_data(dataset_path)
+    cleaned_data = preprocess_data(data)
+    processed_data_path = 'data/processed/processed_dataset.csv'
+    save_processed_data(cleaned_data, processed_data_path)
 
-        # Proceed with model fine-tuning
-        return fine_tune(model_name, dataset_url=None, file=processed_data_path, epochs=epochs, batch_size=batch_size, learning_rate=learning_rate)
-    except Exception as e:
-        print(f"Error occurred: {e}")
-        return str(e)
+    # Proceed with model fine-tuning
+    return fine_tune(model_name, dataset_url=None, file=processed_data_path, epochs=epochs, batch_size=batch_size, learning_rate=learning_rate)
 
 iface = gr.Interface(
     fn=prepare_and_train,
@@ -30,4 +28,4 @@ def prepare_and_train(model_name, dataset_path, epochs, batch_size, learning_rat
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()