Update start.py

start.py

@@ -1,34 +1,18 @@
-# start.py
-# This script installs torch (CPU wheel) at runtime if not present, then launches app.py.
-# Hugging Face Spaces runs pip install from requirements.txt during build; but some packages'
-# setup.py import torch at that time causing failures. We ensure torch exists before importing tinyllava.
-
-import os
-import subprocess
-import sys
-import pkg_resources
-import importlib
+# start.py - ensure torch is installed at runtime then run app.py
+import os, sys, subprocess
 
 def ensure_torch():
     try:
-        import torch  # type: ignore
-        print("torch already installed:", torch.__version__)
+        import torch
+        print("torch present:", torch.__version__)
         return
     except Exception:
-        print("torch not found — installing CPU wheel...")
-
-    # Install CPU-only torch compatible with many Linux builds. Use pip to install the CPU wheel.
-    # Use a stable torch CPU build (version chosen to be recent and broadly compatible).
-    cmd = [sys.executable, "-m", "pip", "install", "--no-cache-dir", "torch==2.2.0+cpu", "--extra-index-url", "https://download.pytorch.org/whl/cpu"]
-    subprocess.check_call(cmd)
-    # verify
-    import torch  # type: ignore
-    print("Installed torch", torch.__version__)
-
-def main():
-    ensure_torch()
-    # Now launch the main app. We exec app.py so that environment is clean and imports happen after torch install.
-    os.execvp(sys.executable, [sys.executable, "app.py"])
+        print("Installing CPU torch...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-cache-dir",
+                               "torch==2.2.0+cpu", "--extra-index-url", "https://download.pytorch.org/whl/cpu"])
+        import torch
+        print("Installed torch", torch.__version__)
 
 if __name__ == "__main__":
-    main()
+    ensure_torch()
+    os.execv(sys.executable, [sys.executable, "app.py"])