Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 from diagrams import Diagram, Cluster
-from diagrams.azure.compute import VirtualNetworks
-from diagrams.azure.network import LoadBalancers, TrafficManagers
+from diagrams.azure.network import VirtualNetwork, LoadBalancer, TrafficManager
 
 def main():
     st.set_page_config(page_title="Azure Network Diagram Demo")
@@ -14,9 +13,9 @@ def main():
 
         with Diagram("Azure Network Diagram", show=False):
             with Cluster("Azure"):
-                vn = VirtualNetworks("Virtual Network")
-                lb = LoadBalancers("Load Balancer")
-                tm = TrafficManagers("Traffic Manager")
+                vn = VirtualNetwork("Virtual Network")
+                lb = LoadBalancer("Load Balancer")
+                tm = TrafficManager("Traffic Manager")
 
             vn >> lb >> tm