sanjeevbora commited on
Commit
bbcf649
·
verified ·
1 Parent(s): 621a2f6

updated for installations

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -12,11 +12,14 @@ import re
12
  import transformers
13
  from torch import bfloat16
14
  from langchain_community.document_loaders import DirectoryLoader
15
- import subprocess
16
-
17
- # Run setup.sh script before starting the app
18
- subprocess.run(["/bin/bash", "setup.sh"], check=True)
19
-
 
 
 
20
  # Initialize embeddings and ChromaDB
21
  model_name = "sentence-transformers/all-mpnet-base-v2"
22
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
12
  import transformers
13
  from torch import bfloat16
14
  from langchain_community.document_loaders import DirectoryLoader
15
+ # import subprocess
16
+ import os
17
+ # # Run setup.sh script before starting the app
18
+ # subprocess.run(["/bin/bash", "setup.sh"], check=True)
19
+
20
+ os.system('apt-get update && apt-get install -y libmagic1')
21
+ os.system('pip install accelerate')
22
+ os.system('pip install -i https://pypi.org/simple/ bitsandbytes')
23
  # Initialize embeddings and ChromaDB
24
  model_name = "sentence-transformers/all-mpnet-base-v2"
25
  device = "cuda" if torch.cuda.is_available() else "cpu"