Update app.py

app.py

@@ -2,13 +2,12 @@ import pandas as pd
 import numpy as np
 from sklearn.ensemble import RandomForestClassifier
 import gradio as gr
-import zipfile
 
 def load_and_train_model():
-    with zipfile.ZipFile('train.csv.zip') as z:
-        with z.open('train.csv') as f:
-            train_data = pd.read_csv(f)
+    # Load data
+    train_data = pd.read_csv('train.csv.zip')
     
+    # Select important features
     important_features = ['Elevation', 'Horizontal_Distance_To_Roadways', 
                          'Horizontal_Distance_To_Fire_Points', 
                          'Horizontal_Distance_To_Hydrology',
@@ -17,6 +16,7 @@ def load_and_train_model():
     X = train_data[important_features]
     y = train_data['Cover_Type']
     
+    # Train model
     model = RandomForestClassifier(n_estimators=100, random_state=42)
     model.fit(X, y)