updated handler.py

handler.py

@@ -1,6 +1,18 @@
 from typing import Dict, List, Any
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import subprocess
+import importlib
 
+def install_and_import(package):
+    try:
+        importlib.import_module(package)
+    except ImportError:
+        subprocess.check_call(["pip", "install", package])
+    finally:
+        globals()[package] = importlib.import_module(package)
+
+# Ensure 'bitsandbytes' is installed
+install_and_import('bitsandbytes')
 class EndpointHandler():
     def __init__(self, path=""):
       model_id = "DisgustingOzil/Academic-MCQ-Generator"