Update app.py

app.py

@@ -1,21 +1,7 @@
+import torch
 from llmtuner import create_ui
-import gradio as gr
 
-css = '''
-#duplicate-button {
-  margin: auto;
-  color: white;
-  background: #1565c0;
-  border-radius: 100vh;
-}
-'''
-title = 'LLaMA Board: A One-stop Web UI for Getting Started with LLaMA Factory'
-description = '<a href="https://github.com/hiyouga/LLaMA-Factory" target="_blank">LLaMA Factory</a>: Training and Evaluating Large Language Models with Minimal Effort'
 
-with gr.Blocks(css=css) as demo:
-  gr.HTML(f"<h1><center>{title}</center></h1>")
-  gr.HTML(f"<h3><center>{description}</center></h3>")
-  gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
-  demo_ui = create_ui()
-
-demo.queue().launch()
+if __name__ == "__main__"
+    demo_ui = create_ui(demo_mode=(not torch.cuda.is_available()))
+    demo.queue().launch()