File size: 536 Bytes
83a5322 f3b414c 83a5322 8396f74 efbb976 83a5322 efbb976 83a5322 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | import gradio as gr
from src.pages.interrogation import build as interrogation_build
from src.pages.logs import build as logs_build
import subprocess
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
interrogation_page = interrogation_build()
logs_page = logs_build()
with gr.Blocks() as app:
gr.TabbedInterface([interrogation_page, logs_page], [
"Interrogation", "Logs"
])
if __name__ == "__main__":
app.queue().launch(show_error=True)
|