Update README.md

README.md

@@ -1,5 +1,5 @@
 ---
-library_name: "torch"
+library_name: torch
 tags:
 - image-classification
 - resnet
@@ -18,7 +18,7 @@ pipeline_tag: image-classification
 datasets:
 - phiyodr/coco2017
 - HuggingFaceM4/ChartQA
-- zhibei1204/DiagramQG
+- JasmineQiuqiu/diagrams_with_captions_2
 ---
 
 # Model Card for Diagram Classification Model
@@ -115,7 +115,7 @@ print(predict("my-diagram-classifier/31188_1536932698.jpg"))
 
 The model was trained using:
 - **ChartQA dataset** (for diagram samples)
-- **DiagramQG dataset** (for diagram samples)
+- **JasmineQiuqiu/diagrams_with_captions_2** (for diagram samples)
 - **COCO dataset (subset)** (for non-diagram samples)
 
 ### Training Procedure