From db9e92176fc973014ffe6fe59f05f222e318633d Mon Sep 17 00:00:00 2001
From: Esteban Cara de Sexo <8595080+canstralian@users.noreply.github.com>
Date: Sun, 2 Feb 2025 15:53:13 +1000
Subject: [PATCH] Update app.py

---
 app.py | 22 ++++++++++------------
 1 file changed, 10 insertions(+), 12 deletions(-)

diff --git a/app.py b/app.py
index 82b09e4..a39f41f 100644
--- a/app.py
+++ b/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()
\ No newline at end of file
+    iface.launch()