Update pages/2_CodeSimulator.py

pages/2_CodeSimulator.py

@@ -25,7 +25,18 @@ def execute_code(code):
 st.title('PyTorch Code Runner')
 
 # Text area for inputting the PyTorch code
-code_input = st.text_area("Enter your PyTorch code here", height=300)
+code_input = st.text_area("Enter your PyTorch code here", height=300, value="""# Create two tensors of different shapes
+tensor_c = torch.tensor([1, 2, 3])
+tensor_d = torch.tensor([[1], [2], [3]])
+
+# Perform addition using broadcasting
+tensor_broadcast_add = tensor_c + tensor_d
+print("Broadcast Addition:\\n", tensor_broadcast_add)
+
+# Perform element-wise multiplication using broadcasting
+tensor_broadcast_mul = tensor_c * tensor_d
+print("Broadcast Multiplication:\\n", tensor_broadcast_mul)
+""")
 
 # Button to execute the code
 if st.button("Run Code"):