| 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) |
|
|
