Spaces:
Build error
Build error
PlasmarineXD
commited on
Commit
·
1382180
1
Parent(s):
b7b9b8e
Re-init
Browse files- app.py +0 -13
- install_dependencies.sh +0 -10
- requirements.txt +1 -1
- 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
|
|
|
|
|
|
|
|
|
|