<feat> using app.sh.

README.md

@@ -5,7 +5,7 @@ colorFrom: green
 colorTo: gray
 sdk: gradio
 sdk_version: 5.33.2
-app_file: app.py
+app_file: app.sh
 pinned: false
 license: apache-2.0
 short_description: huggingface space for DRA-Ctrl.

app.py

@@ -12,6 +12,8 @@ import gc
 import gradio as gr
 import numpy as np
 
+from flash_attn import flash_attn_func
+
 from peft import LoraConfig
 from omegaconf import OmegaConf
 from safetensors.torch import safe_open

app.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+set -e
+
+apt-get update && apt-get install -y git build-essential python3-dev ninja-build
+
+pip install -r requirements.txt
+pip install flash-attn==2.7.3 --no-build-isolation
+
+python app.py