Update app.py

app.py

@@ -1,21 +1,15 @@
 import gradio as gr
 import pandas as pd
-import mlflow
 import numpy as np
 import os
+import joblib  # Change to joblib for direct loading
 
 # Load the model - for Hugging Face deployment
 try:
-    # Get absolute path to current directory
-    current_dir = os.getcwd()
-    print(f"Loading model from: {current_dir}")
-    
-    # List all files to verify
-    print("Available files:", os.listdir(current_dir))
-    
-    # Load model from current directory
-    model = mlflow.sklearn.load_model(current_dir)
+    # Load model directly from python_model.pkl
+    model = joblib.load("python_model.pkl")
     print("Model loaded successfully")
+    print("Model type:", type(model))
     
 except Exception as e:
     print(f"Error loading model: {str(e)}")