Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import pandas as pd
 
-MODELS = ['pythia-12B', 'pythia-6.9B', 'pythia-1.4B', 'pythia-410M']
+MODELS = ['pythia-12B', 'pythia-6.7B', 'pythia-1.4B', 'pythia-410M']
 
 def load_csv(model_name):
     csv_path = f"model_csvs/{model_name}.csv"
@@ -26,9 +26,33 @@ def update_csv_dropdown(model_name):
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-        # Rethinking LLM Memorization through the Lens of Adversarial Compression
+        <div style="background-color: #B22234; padding: 20px; text-align: center; font-size: 24px; color: white; font-weight: bold;">
+            Rethinking LLM Memorization through the Lens of Adversarial Compression
+        </div>
+        
+        <div style="background-color: #B22234; padding: 10px; text-align: center; font-size: 18px; color: blanchedalmond;">
+            <a href="https://www.avischwarzschild.com/" style="color: blanchedalmond; text-decoration: none; font-weight: bold;">Avi Schwarzschild</a><sup style="color: blanchedalmond;">*</sup>&nbsp;&nbsp;&nbsp;
+            <a href="https://zhilif.github.io/" style="color: blanchedalmond; text-decoration: none; font-weight: bold;">Zhili Feng</a><sup style="color: blanchedalmond;">*</sup>&nbsp;&nbsp;&nbsp;
+            <a href="https://pratyushmaini.github.io/" style="color: blanchedalmond; text-decoration: none; font-weight: bold;">Pratyush Maini</a>&nbsp;&nbsp;&nbsp;
+            <a href="https://acmilab.org/" style="color: blanchedalmond; text-decoration: none; font-weight: bold;">Zack Lipton</a>&nbsp;&nbsp;&nbsp;
+            <a href="https://zicokolter.com/" style="color: blanchedalmond; text-decoration: none; font-weight: bold;">Zico Kolter</a>
+        </div>
+
+        <div style="background-color: #B22234; padding: 10px; text-align: center; font-size: 18px; color: white;">
+        Carnegie Mellon University
+        </div>
+
+
+        <div style="padding: 0px; text-align: center; align: center;">
+        <br>
+            <a href="https://github.com/locuslab/memorization-in-llms/" style="color: white; text-decoration: none;  text-align: center; align: center;">
+                <img src="https://img.shields.io/badge/GitHub-Repository-blue?style=flat-square&logo=github&logoColor=white" alt="GitHub Repository" style="color: white; text-decoration: none;  text-align: center; align: center;">
+            </a>
+        </div>
 
-        Authors: Avi Schwarzschild\*, Zhili Feng\*, Pratyush Maini, Zack Lipton, Zico Kolter
+        
+        
+        </div>
 
         ## Abstract
 
@@ -96,7 +120,7 @@ with gr.Blocks() as demo:
         If you find this work useful, please consider citing our paper:
 
         ```bibtex
-        @article{schwarzschild2023rethinking,
+        @article{schwarzschild2024rethinking,
           title={Rethinking LLM Memorization through the Lens of Adversarial Compression},
           author={Schwarzschild, Avi and Feng, Zhili and Maini, Pratyush and Lipton, Zack and Kolter, Zico},
           journal={arXiv preprint},