Update app.py

app.py

@@ -7,15 +7,13 @@ import subprocess
 import sys
 
 def install_bitsandbytes():
-    # Uninstall bitsandbytes if installed
-    subprocess.run([sys.executable, "-m", "pip", "uninstall", "-y", "bitsandbytes"])
-    
-    # Reinstall bitsandbytes with the required options
-    subprocess.run([sys.executable, "-m", "pip", "install", "bitsandbytes", "--force-reinstall", "--index-url", "https://pypi.org/simple/", "--no-cache-dir"])
+    # Reinstall bitsandbytes with CPU support
+    subprocess.run([sys.executable, "-m", "pip", "install", "bitsandbytes-cpu", "--force-reinstall", "--index-url", "https://pypi.org/simple/", "--no-cache-dir"])
 
 # Call the function at the start of the script
 install_bitsandbytes()
 
+
 # Load the model and processor
 repo_name = "cyan2k/molmo-7B-O-bnb-4bit"
 arguments = {"device_map": "auto", "torch_dtype": "auto", "trust_remote_code": True}