Spaces:
Build error
Build error
wissamantoun
commited on
Commit
•
412c787
1
Parent(s):
0558cbb
adding git lfs
Browse files- app.py +4 -1
- backend/utils.py +9 -0
- 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
|