Update app.py

app.py

@@ -1,7 +1,7 @@
 """
 ShukGEN v3 — Flask Backend Server (Hugging Face Spaces Edition)
 Serves both the UI (index.html) and the API on port 7860.
-Model is auto-loaded from model.pt in the same directory at startup.
+Model is auto-loaded from shukgen_v3_final.pth in the same directory at startup.
 """
 
 import io, base64, os, sys, traceback, threading, time, math
@@ -557,15 +557,15 @@ def api_save_all():
 
 # ─────────────────────────────────────────────────────────────────────────────
 #  AUTO-LOAD MODEL ON STARTUP
-#  Place your model file as  model.pt  in the same directory as this script.
+#  Place your model file as  shukgen_v3_final.pth  in the same directory as this script.
 #  It will be loaded automatically when the Space starts.
 # ─────────────────────────────────────────────────────────────────────────────
 
-MODEL_PATH = os.path.join(BASE_DIR, 'model.pt')
+MODEL_PATH = os.path.join(BASE_DIR, 'shukgen_v3_final.pth')
 
 def auto_load_model():
     if os.path.isfile(MODEL_PATH):
-        print(f"  🔍  Found model.pt — loading...")
+        print(f"  🔍  Found shukgen_v3_final.pth — loading...")
         try:
             model, image_size, cfg = load_model(MODEL_PATH, STATE['device'])
             STATE['model']        = model
@@ -576,8 +576,8 @@ def auto_load_model():
             print(f"  ❌  Failed to auto-load model: {e}")
             traceback.print_exc()
     else:
-        print("  ⚠️   No model.pt found in app directory.")
-        print("       Upload model.pt to the Space repo and restart.")
+        print("  ⚠️   No shukgen_v3_final.pth found in app directory.")
+        print("       Upload shukgen_v3_final.pth to the Space repo and restart.")
 
 
 if __name__ == '__main__':