Update app.py

app.py

@@ -12,7 +12,10 @@ def ensure_library_installed(library_name, install_name=None):
     except ImportError:
         install_name = install_name or library_name
         subprocess.check_call(["pip", "install", install_name])
-        globals()[library_name] = __import__(library_name)
+        try:
+            globals()[library_name] = __import__(library_name)  # Retry import
+        except ImportError as e:
+            raise ImportError(f"Failed to import {library_name} after installation.") from e
 
 # Ensure required libraries
 ensure_library_installed("torch")