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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|