PlasmarineXD commited on
Commit
1382180
·
1 Parent(s): b7b9b8e
Files changed (4) hide show
  1. app.py +0 -13
  2. install_dependencies.sh +0 -10
  3. requirements.txt +1 -1
  4. setup.sh +0 -4
app.py CHANGED
@@ -1,17 +1,4 @@
1
- import subprocess
2
  import os
3
-
4
- # Function to install packages
5
- def install(package):
6
- subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
7
-
8
- # Install necessary packages
9
- install("torch")
10
- install("flash-attn")
11
- install("streamlit")
12
- install("transformers")
13
-
14
- # Import installed packages
15
  import streamlit as st
16
  from transformers import AutoModel, AutoTokenizer
17
 
 
 
1
  import os
 
 
 
 
 
 
 
 
 
 
 
 
2
  import streamlit as st
3
  from transformers import AutoModel, AutoTokenizer
4
 
install_dependencies.sh DELETED
@@ -1,10 +0,0 @@
1
- #!/bin/bash
2
-
3
- # Ensure pip is up to date
4
- pip install --upgrade pip
5
-
6
- # Install torch first
7
- pip install torch
8
-
9
- # Install other dependencies
10
- pip install streamlit transformers flash-attn
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  torch
2
  streamlit
3
  transformers
4
- flash-attn
 
1
  torch
2
  streamlit
3
  transformers
4
+ flash-attn
setup.sh DELETED
@@ -1,4 +0,0 @@
1
- #!/bin/bash
2
-
3
- # Run the installation script
4
- ./install_dependencies.sh