Update README.md

README.md

@@ -1,3 +1,47 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+datasets:
+- nllg/datikz-v2
+---
+
+# Model Card for DeTi*k*Zify-TL<sub>1.1b</sub>
+DeTi*k*Zify-TL<sub>1.1b</sub> is a language model that automatically converts
+sketches and existing scientific figures into editable, semantics-preserving
+Ti*k*Z graphics programs. It is based on
+[TinyLlama](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T)
+and was fine-tuned on
+[DaTi*k*Z<sub>v2</sub>](https://huggingface.co/datasets/nllg/datikz-v2). Check
+out the [DeTi*k*Zify](https://github.com/potamides/DeTikZify) project for more
+information and tips on how to best run the model.
+
+## Usage
+```python
+from operator import itemgetter
+
+from detikzify.model import load
+from detikzify.infer import DetikzifyPipeline
+import torch
+
+image = "https://w.wiki/A7Cc"
+pipeline = DetikzifyPipeline(*load(
+    base_model="nllg/detikzify-tl-1.1b",
+    device_map="auto",
+    torch_dtype=torch.bfloat16,
+))
+
+# generate a single TikZ program
+fig = pipeline.sample(image=image)
+
+# if it compiles, rasterize it and show it
+if fig.is_rasterizable:
+    fig.rasterize().show()
+
+# run MCTS for 10 minutes and generate multiple TikZ programs in the process
+figs = set()
+for score, fig in pipeline.simulate(image=image, timeout=600):
+    figs.add((score, fig))
+
+# save the best TikZ program
+best = sorted(figs, key=itemgetter(0))[-1][1]
+best.save("fig.tex")
+```