update

app.py

@@ -44,7 +44,7 @@ class TimeSeriesEditor:
         self.freq_bands = np.ones(5)  # 5 frequency bands, initially all set to 1.0
         self.function_parser = FunctionParser()
         self.trending_controls = [
-            (200, 250, 0, self.function_parser.string_to_function("sin(2*pi*x)"), 0.05)
+            # (200, 250, 0, self.function_parser.string_to_function("sin(2*pi*x)"), 0.05)
             # 200,250,0,sin(2*pi*x),0.05
         ]
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -454,6 +454,9 @@ class TimeSeriesEditor:
                 # model_control_signal=model_control_signal,
                 gradient_control_signal=gradient_control_signal
             )
+            
+            observed_points = observed_points.cpu()
+            observed_mask = observed_mask.cpu()
 
         # Store latest results
         self.latest_sample = sample
@@ -619,7 +622,7 @@ class TimeSeriesEditor:
 def create_gradio_interface(editor: TimeSeriesEditor):
     with gr.Blocks() as app:
         gr.Markdown("# Time Series Editor")
-        gr.Markdown("## Instruction: Scroll Down + Click [Update Figure] [~20s]")
+        gr.Markdown("## Instruction: Scroll Down + Click [Update Figure] [~10s]")
 
         metrics_display = gr.JSON(label="Metrics", value={})
 
@@ -658,7 +661,7 @@ def create_gradio_interface(editor: TimeSeriesEditor):
                         headers=["time", "feature", "value"],
                         datatype=["number", "number", "number"],
                         # label="Anchor Point Control",
-                        value=[[0, 0, 0.04], [2, 0, 0.58], [6, 0, 0.27], [58, 0, 1.0], [60, 0, 0.5]],
+                        value=[[0, 0, 0.04], [2, 0, 0.58], [6, 0, 0.27], [58, 0, 0.8], [60, 0, 0.5]],
                         col_count=(3, "fixed"),  # Fix number of columns
                         interactive=True
                     )
@@ -713,7 +716,7 @@ def create_gradio_interface(editor: TimeSeriesEditor):
                     
                     Separate multiple trends with semicolons.
                     """)
-                    enable_trending_control = gr.Checkbox(label="Enable Trending Control", value=True)
+                    enable_trending_control = gr.Checkbox(label="Enable Trending Control", value=False)
                     enable_trending_control_with_diff = gr.Checkbox(label="Consider Last Generated", value=False)
                     trending_control = gr.Textbox(
                         label="Trending Control Parameters",