Spaces:

syedfaisalabrar
/

License_Classification

Sleeping

syedfaisalabrar commited on Feb 26

Commit

7eebf99

verified ·

1 Parent(s): 9ae840f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ model_path = "best.pt"
 model = YOLO(model_path)
 os.system("pip uninstall -y bitsandbytes")
-os.system("pip install --force-reinstall 'https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_multi-backend-refactor/bitsandbytes-0.44.1.dev0-py3-none-win_amd64.whl'")
 os.system("pip install intel_extension_for_pytorch")
 os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Forces CPU mode

 model = YOLO(model_path)
 os.system("pip uninstall -y bitsandbytes")
+os.system("pip install --force-reinstall 'https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_multi-backend-refactor/bitsandbytes-0.44.1.dev0-py3-none-any.whl'")
 os.system("pip install intel_extension_for_pytorch")
 os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Forces CPU mode