asfdaf

handler.py

@@ -6,23 +6,21 @@ import importlib
 import subprocess
 import importlib
 
-def install_package(package):
-    subprocess.check_call(["pip", "install", package])
+import subprocess
+import importlib
 
-def install_and_import(package, index_url=None):
+def install_and_import(package):
     try:
         importlib.import_module(package)
     except ImportError:
-        if index_url:
-            subprocess.check_call(["pip", "install", package, "--index-url", index_url])
-        else:
-            subprocess.check_call(["pip", "install", package])
+        subprocess.check_call(["pip", "install", package])
     finally:
         globals()[package] = importlib.import_module(package)
 
-# Install packages
-install_package('xformers --index-url https://download.pytorch.org/whl/cu121')
-install_package('unsloth[kaggle-new] @ git+https://github.com/unslothai/unsloth.git')
+# Ensure 'bitsandbytes' is installed
+install_and_import('accelerate')
+install_and_import('bitsandbytes')
+
 class EndpointHandler():
     def __init__(self, path=""):
       model_id = "DisgustingOzil/Academic-MCQ-Generator"