Fix bugs

app.py

@@ -13,20 +13,21 @@ from transformers import (
 
 # ─── MODEL SETUP ────────────────────────────────────────────────────────────────
 # Default to 8B but keep both variants resident on the GPU.
-DEFAULT_MODEL_NAME = "bytedance-research/ChatTS-8B"
+DEFAULT_MODEL_NAME = "ChatTS-8B"
 AVAILABLE_MODEL_NAMES = [
-    "bytedance-research/ChatTS-8B",
-    "bytedance-research/ChatTS-14B"
+    "ChatTS-8B",
+    "ChatTS-14B"
 ]
 
 MODEL_REGISTRY = {}
 
 for name in AVAILABLE_MODEL_NAMES:
     print(f"Loading model into memory: {name}")
-    tok = AutoTokenizer.from_pretrained(name, trust_remote_code=True)
-    proc = AutoProcessor.from_pretrained(name, trust_remote_code=True, tokenizer=tok)
+    model_path = "bytedance-research/" + name
+    tok = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
+    proc = AutoProcessor.from_pretrained(model_path, trust_remote_code=True, tokenizer=tok)
     mdl = AutoModelForCausalLM.from_pretrained(
-        name,
+        model_path,
         trust_remote_code=True,
         device_map="auto",
         torch_dtype=torch.float16
@@ -66,8 +67,9 @@ def load_model_by_name(name: str):
 
 
 def switch_model(selected_model_name: str):
-    """Wrapper for Gradio to switch models; returns status text."""
-    return load_model_by_name(selected_model_name)
+    """Wrapper for Gradio to switch models via radio selection."""
+    load_model_by_name(selected_model_name)
+    return None
 
 # ─── HELPER FUNCTIONS ──────────────────────────────────────────────────────────
 
@@ -332,27 +334,20 @@ with gr.Blocks(title="ChatTS Demo") as demo:
         with gr.Column(scale=1):
             # Model selection UI
             model_radio = gr.Radio(
-                choices=["bytedance-research/ChatTS-8B", "bytedance-research/ChatTS-14B"],
+                choices=["ChatTS-8B", "ChatTS-14B"],
                 value=CURRENT_MODEL_NAME,
                 label="Model Version"
             )
 
-            model_btn = gr.Button("Load Model")
-
-            model_status = gr.Textbox(
-                label="Model Status",
-                value=f"Models in memory: {', '.join(AVAILABLE_MODEL_NAMES)}. Active: {CURRENT_MODEL_NAME}",
-                interactive=False
-            )
-
             upload = gr.File(
                 label="Upload CSV File",
                 file_types=[".csv"],
-                type="filepath"
+                type="filepath",
+                height=120
             )
             
             prompt_input = gr.Textbox(
-                lines=6,
+                lines=5,
                 placeholder="Enter your question here...",
                 label="Analysis Prompt",
                 value="Please analyze all the given time series and provide insights about the local fluctuations in the time series in detail."
@@ -362,11 +357,11 @@ with gr.Blocks(title="ChatTS Demo") as demo:
             
         with gr.Column(scale=2):
             series_selector = gr.Dropdown(
-                label="Select a Column to Visualize",
+                label="Select a Channel to Visualize (All Channels Will be Input to ChatTS)",
                 choices=[],
                 value=None
             )
-            plot_out = gr.LinePlot(value=pd.DataFrame(), label="Time Series Visualization")
+            plot_out = gr.LinePlot(value=pd.DataFrame(), label="Channel Visualization (All Channels Will be Input to ChatTS)")
             file_status = gr.Textbox(
                 label="File Status", 
                 interactive=False,
@@ -410,11 +405,11 @@ with gr.Blocks(title="ChatTS Demo") as demo:
         outputs=[text_out]
     )
 
-    # Wire model loading button
-    model_btn.click(
+    # Model selection reacts immediately; no separate button needed
+    model_radio.change(
         fn=switch_model,
         inputs=[model_radio],
-        outputs=[model_status]
+        outputs=[]
     )
 
 if __name__ == '__main__':