HyperX-Sen commited on
Commit
bfdc0f0
·
verified ·
1 Parent(s): 2551572

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ def install(package):
6
  subprocess.check_call([sys.executable, "-m", "pip", "install", package])
7
 
8
  # List of required libraries
9
- required_packages = ["transformers", "torch", "accelerate", "gradio", "huggingface_hub"]
10
 
11
  # Install any missing packages
12
  for package in required_packages:
 
6
  subprocess.check_call([sys.executable, "-m", "pip", "install", package])
7
 
8
  # List of required libraries
9
+ required_packages = ["transformers", "torch", "accelerate", "gradio", "huggingface_hub", "safetensors"]
10
 
11
  # Install any missing packages
12
  for package in required_packages: