Update app.py

app.py

@@ -72,7 +72,6 @@ def compare_algorithms(code1, code2):
     buf.seek(0)
     return Image.open(buf)
 
-# Gradio interface
 interface = gr.Interface(
     fn=compare_algorithms,
     inputs=[
@@ -81,7 +80,14 @@ interface = gr.Interface(
     ],
     outputs=gr.Image(type="pil", label="Token Embedding PCA"),
     title="GraphCodeBERT Token Embedding Comparison",
-    description="Edit or paste two Python code snippets. This tool compares their token-level embeddings using GraphCodeBERT and PCA."
+    description="Edit or paste two Python code snippets. This tool compares their token-level embeddings using GraphCodeBERT and PCA.",
+    article="""
+**Citation**  
+Martinez-Gil, J. (2025). *Augmenting the Interpretability of GraphCodeBERT for Code Similarity Tasks.* International Journal of Software Engineering and Knowledge Engineering, 35(05), 657–678.
+
+**GitHub Repository**  
+[View Source on GitHub](https://github.com/jorge-martinez-gil/graphcodebert-interpretability)
+"""
 )
 
 if __name__ == "__main__":