Add flash atttention to requirements file

app.py

@@ -166,8 +166,8 @@ def process_and_visualize(text, model):
 
 
 
-with gr.Blocks(title="Phi-3 Mini 4k Instruct Graph (by Emergent Methods") as demo:
-    gr.Markdown("# Phi-3 Mini 4k Instruct Graph (by Emergent Methods)")
+with gr.Blocks(title="Phi-3 Instruct Graph (by Emergent Methods") as demo:
+    gr.Markdown("# Phi-3 Instruct Graph (by Emergent Methods)")
     gr.Markdown("Extract a JSON graph from a text input and visualize it.")
     with gr.Row():
         with gr.Column(scale=1):

requirements.txt

@@ -8,3 +8,4 @@ spaces
 pyvis 
 networkx
 spacy
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl