Add installation script to app.py and change install_dependencies

app.py

@@ -1,16 +1,17 @@
-import os
 import subprocess
+import os
 
-# Install dependencies if not present
+# Function to install packages
 def install(package):
     subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
 
+# Install necessary packages
 install("torch")
+install("flash-attn")
 install("streamlit")
 install("transformers")
-install("flash-attn")
 
-# Your existing imports and code
+# Import installed packages
 import streamlit as st
 from transformers import AutoModel, AutoTokenizer
 

install_dependencies.sh

@@ -1,5 +1,8 @@
 #!/bin/bash
 
+# Ensure pip is up to date
+pip install --upgrade pip
+
 # Install torch first
 pip install torch