#!/bin/bash

# Install torch first
pip install torch

# Install other dependencies
pip install streamlit transformers flash-attn