Spaces:
Running
Running
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
|