Update app.py

app.py

@@ -9,27 +9,32 @@ def log_tensor(name, x):
 
 # Simple function
 def g(x, y):
-    log_tensor("g input x", x)
-    log_tensor("g input y", y)
     z = x + y
-    log_tensor("g output z", z)
     return z
 
 # Compiled function
 @torch.compile(backend="eager")
 def f(x):
-    log_tensor("f input x", x)
     x = torch.sin(x)
-    log_tensor("f after torch.sin(x)", x)
     x = g(x, x)
-    log_tensor("f after g(x, x)", x)
     return x
 
-# Example input
-x = torch.ones(3, 3, dtype=torch.float32)
-log_tensor("original input x", x)
+# --- Initialization / run once ---
+def init_and_run_once():
+    # Example input
+    x = torch.ones(3, 3, dtype=torch.float32)
+    print("=== INITIAL INPUT ===")
+    log_tensor("original input x", x)
 
-# Run compiled function
-out = f(x)
+    # Run compiled function once
+    out = f(x)
 
-log_tensor("final output", out)
+    print("=== OUTPUT AFTER FIRST RUN ===")
+    log_tensor("final output", out)
+
+    # Return output if needed
+    return out
+
+# Run once at import / init
+if __name__ == "__main__":
+    init_and_run_once()