Update app.py

app.py

@@ -106,8 +106,8 @@ def main():
     
     with tab1:
         st.subheader("Architecture Diagram")
-        st.image("https://upload.wikimedia.org/wikipedia/commons/thumb/8/8a/Transformer_model.svg/1200px-Transformer_model.svg.png", 
-                use_container_width=True)  # Changed parameter here
+        st.image("https://jalammar.github.io/images/t/transformer.png",
+                  use_container_width=True)
     
     with tab2:
         st.subheader("Model Size Comparison")