Update README.md

README.md

@@ -34,6 +34,10 @@ Object Detection results format:
 {'\<OD>': {'bboxes': [[x1, y1, x2, y2], ...], 
 'labels': ['label1', 'label2', ...]} }
 
+## Training Code and Dataset
+- Dataset: [yifeihu/TF-ID-arxiv-papers](https://huggingface.co/datasets/yifeihu/TF-ID-arxiv-papers)
+- Code: [github.com/ai8hyf/TF-ID](https://github.com/ai8hyf/TF-ID)
+
 ## Benchmarks
 
 We tested the models on paper pages outside the training dataset. The papers are a subset of huggingface daily paper.
@@ -88,18 +92,15 @@ print(parsed_answer)
 
 To visualize the results, see [this tutorial notebook](https://colab.research.google.com/github/roboflow-ai/notebooks/blob/main/notebooks/how-to-finetune-florence-2-on-detection-dataset.ipynb) for more details.
 
-## Finetuning Code and Dataset
-
-Dataset: [yifeihu/TF-ID-arxiv-papers](https://huggingface.co/datasets/yifeihu/TF-ID-arxiv-papers)
-
-Code: Coming soon!
-
 ## BibTex and citation info
 
 ```
-@misc{TF-ID, 
-      url={[https://huggingface.co/yifeihu/TF-ID-base](https://huggingface.co/yifeihu/TF-ID-base)}, 
-      title={TF-ID: Table/Figure IDentifier for academic papers}, 
-      author={"Yifei Hu"}
+@misc{TF-ID,
+  author = {Yifei Hu},
+  title = {TF-ID: Table/Figure IDentifier for academic papers},
+  year = {2024},
+  publisher = {GitHub},
+  journal = {GitHub repository},
+  howpublished = {\url{https://github.com/ai8hyf/TF-ID}},
 }
 ```