dwb2023 commited on
Commit
9fedee1
·
verified ·
1 Parent(s): 5f2d3f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,6 +3,9 @@ import os
3
  from repo_utils import extract_repo_content
4
  from model_utils import get_model_summary, install_flash_attn
5
 
 
 
 
6
  def format_output(extracted_content, repo_url):
7
  formatted_output = f"# Repository URL: {repo_url}\n\n"
8
  for file_data in extracted_content:
 
3
  from repo_utils import extract_repo_content
4
  from model_utils import get_model_summary, install_flash_attn
5
 
6
+ # Install required package
7
+ install_flash_attn()
8
+
9
  def format_output(extracted_content, repo_url):
10
  formatted_output = f"# Repository URL: {repo_url}\n\n"
11
  for file_data in extracted_content: