update app

app.py

@@ -42,13 +42,21 @@ datasets = {}
 pipelines = {}
 
 if selected_task == "Pretraining dataset":
-    st.title("Pretraining Datasets 📚")
+    st.title("Pretraining datasets 📚")
     for model in selected_models:
         with open(f"datasets/{model.lower()}.txt", "r") as f:
             text = f.read()
-        #datasets[model] = text 
         st.markdown(f"## {model}:")
         st.markdown(text)
+        
+elif selected_task == "Model architecture":
+    st.title("Model architecture 🛠")
+    for model in selected_models:
+        with open(f"architectures/{model.lower()}.txt", "r") as f:
+            text = f.read()
+        st.markdown(f"## {model}:")
+        st.markdown(text)
+        
 elif selected_task == "Prompting":
     for model in selected_models:
         if model == "CodeParrot":