Spaces:

dwb2023
/

hf_extractor

Running on Zero

dwb2023 commited on Jun 15

Commit

5f2d3f0

•

1 Parent(s): f480be7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 from repo_utils import extract_repo_content
-from model_utils import get_model_summary
 def format_output(extracted_content, repo_url):
     formatted_output = f"# Repository URL: {repo_url}\n\n"

 import gradio as gr
 import os
 from repo_utils import extract_repo_content
+from model_utils import get_model_summary, install_flash_attn
 def format_output(extracted_content, repo_url):
     formatted_output = f"# Repository URL: {repo_url}\n\n"