Update app.py

app.py

@@ -145,7 +145,24 @@ def create_interface_custom():
     return interface
 
 
+def download_models():
+    import huggingface_hub
+
+    repo_name = "SuperSecureHuman/xray-reg-models"
+    local_dir = "./"
+    huggingface_hub.snapshot_download(
+        repo_id=repo_name,
+        local_dir=local_dir,
+    )
+
+
 def main():
+    # Download models if not already present
+    try:
+        download_models()
+    except Exception as e:
+        logging.error("Error downloading models: %s", e)
+        exit(1)
     logging.basicConfig(level=logging.INFO)
     interface = create_interface_custom()
     interface.launch()
@@ -153,3 +170,7 @@ def main():
 
 if __name__ == "__main__":
     main()
+
+
+if __name__ == "__main__":
+    main()