wissamantoun commited on
Commit
412c787
1 Parent(s): 0558cbb

adding git lfs

Browse files
Files changed (3) hide show
  1. app.py +4 -1
  2. backend/utils.py +9 -0
  3. requirements.txt +2 -1
app.py CHANGED
@@ -1,11 +1,14 @@
1
  import awesome_streamlit as ast
2
  import streamlit as st
3
 
 
 
 
 
4
  import backend.aragpt
5
  import backend.home
6
  import backend.processor
7
  import backend.sa
8
- from backend.utils import get_current_ram_usage
9
 
10
  st.set_page_config(
11
  page_title="TEST", page_icon="📖", initial_sidebar_state="expanded", layout="wide"
 
1
  import awesome_streamlit as ast
2
  import streamlit as st
3
 
4
+ from backend.utils import get_current_ram_usage, install_git_lfs
5
+
6
+ install_git_lfs()
7
+
8
  import backend.aragpt
9
  import backend.home
10
  import backend.processor
11
  import backend.sa
 
12
 
13
  st.set_page_config(
14
  page_title="TEST", page_icon="📖", initial_sidebar_state="expanded", layout="wide"
backend/utils.py CHANGED
@@ -1,5 +1,6 @@
1
  import psutil
2
  from huggingface_hub import Repository
 
3
 
4
 
5
  def get_current_ram_usage():
@@ -14,3 +15,11 @@ def download_models(models):
14
  model, clone_from=f"https://huggingface.co/researchaccount/{model}"
15
  )
16
  return model_dirs
 
 
 
 
 
 
 
 
 
1
  import psutil
2
  from huggingface_hub import Repository
3
+ import os
4
 
5
 
6
  def get_current_ram_usage():
 
15
  model, clone_from=f"https://huggingface.co/researchaccount/{model}"
16
  )
17
  return model_dirs
18
+
19
+
20
+ def install_git_lfs():
21
+ os.system(
22
+ "curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash"
23
+ )
24
+ os.system("sudo apt-get install git-lfs")
25
+ os.system("git lfs install")
requirements.txt CHANGED
@@ -9,4 +9,5 @@ torch==1.9.0
9
  transformers==4.10.0
10
  psutil==5.8.0
11
  fuzzysearch==0.7.3
12
- more-itertools==8.9.0
 
 
9
  transformers==4.10.0
10
  psutil==5.8.0
11
  fuzzysearch==0.7.3
12
+ more-itertools==8.9.0
13
+ cookiecutter