Update app.py

app.py

@@ -1,15 +1,17 @@
 import os
 import subprocess
 import gradio as gr
-from predict import predict_transaction
 
-# Ensure the model directory exists and model is trained
+# Ensure model is trained
 if not (os.path.exists("model/model.pkl") and os.path.exists("model/encoders.pkl")):
     print("Model files not found. Training model...")
     os.makedirs("model", exist_ok=True)
     subprocess.run(["python", "train.py"], check=True)
 
-# Gradio UI wrapper
+# ✅ Import AFTER model is guaranteed to exist
+from predict import predict_transaction
+
+# Gradio UI
 def predict_ui(check_id, employee_id, total, discount_amount, item_count, time, terminal_id):
     return predict_transaction({
         "check_id": check_id,