Update README.md

README.md

@@ -9,21 +9,6 @@ datasets:
 metrics:
 - exact_match
 - rouge
-model-index:
-- name: calc-flan-xl
-  results:
-  - task:
-      type: question-answering
-      name: Question Answering
-    dataset:
-      type: gsm8k
-      name: GSM8K
-      split: validation
-    metrics:
-    - type: exact_match
-      value: 0.495
-    - type: rouge
-      value: 0.655
 license: apache-2.0
 language:
 - en
@@ -31,15 +16,10 @@ language:
 
 # Model Card for calc-flan-xl
 
-<!-- Provide a quick summary of what the model is/does. -->
-
 This model generates reasoning chains over mathematical questions while **using an external tool: Sympy calculator**.
 
-## Model Details
-
-### Model Description
 
-<!-- Provide a longer summary of what this model is. -->
+## Model Description
 
 With the idea to offload a symbolic reasoning from the stochastic language model, 
 we train this model to utilize a calculator **for all applicable numeric operations**.
@@ -52,17 +32,19 @@ This is achieved by training the model to construct calls to the tool's API in t
 where `<gadget>` segment triggers a call of the tool, 
 which is subsequently served by extending model's decoder input context by adding the output of the tool within the `<output>` segment.
 
-- **Developed by:** Anonymous
+- **Developed by:** Calcformer team
 - **Model type:** Autoregressive Encoder-Decoder
 - **Language(s):** en
 - **Finetuned from:** google/flan-t5-xl
 
-### Model Sources
 
-<!-- Provide the basic links for the model. -->
+## Sources
+
+- **Repository:** <https://github.com/prompteus/calc-x>
+- **Paper:** <https://arxiv.org/abs/2305.15017>
+- [**Calcformer model family on HF**](https://huggingface.co/collections/MU-NLPC/calcformers-65367392badc497807b3caf5)
+- [**Calc-X dataset collection on HF**](https://huggingface.co/collections/MU-NLPC/calc-x-652fee9a6b838fd820055483)
 
-- **Repository:** https://github.com/emnlp2023sub/gadgets
-- **Paper:** Stay tuned!
 
 ## Usage
 
@@ -71,20 +53,20 @@ Additionally to conventional generation, using Tool-augmented generation require
 (2) a customization of generate() method augmenting input context on-demand with the outputs of the tools.
 
 You can find these two components implemented in the attached **gadgets/model.py** and **gadgets/gadget.py** in this model's repo 
-and the project's [home repo](https://github.com/emnlp2023sub/gadgets).
+and the project's [home repo](https://github.com/prompteus/calc-x).
 
 After adding these two scripts to your directory, you can use the model as follows:
 
-```python
-
+```python3
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 
 from gadgets.model import gadget_assisted_model
 from gadgets.gadget import Calculator
 
 GadgetAssistedT5 = gadget_assisted_model(T5ForConditionalGeneration)
-model = GadgetAssistedT5.from_pretrained("emnlp2023/calc-flan-xl")
-tokenizer = T5Tokenizer.from_pretrained("emnlp2023/calc-flan-xl")
+model_name = "MU-NLPC/calcformer-flan-xl"
+model = GadgetAssistedT5.from_pretrained(model_name)
+tokenizer = T5Tokenizer.from_pretrained(model_name)
 
 model.prepare_for_generate(tokenizer, 
                            enabled_gadgets=[Calculator()], 
@@ -100,7 +82,9 @@ inputs = tokenizer(query, return_tensors="pt")
 output_ids = model.generate(**inputs)
 tokenizer.decode(output_ids[0], spaces_between_special_tokens=False)
 ```
+
 This returns:
+
 ```html
 According to the ratio, for every 5 parts that Johnson gets, Mike gets 2 parts Since Johnson got $2500,
 each part is therefore $2500/5 = $<gadget id="calculator">2500/5</gadget><output>500</output> 500
@@ -109,26 +93,17 @@ After buying the shirt he will have $1000-$200 = $<gadget id="calculator">1000-2
 Final result is<result>800</result></s>
 ```
 
-### Out-of-Scope Usage
-
-<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+## Out-of-Scope Usage
 
 Note that given the limited scope of the exercises' complexity in the training, this model will not work well for tasks requiring 
 more complex algebraic operations, including equations, variables and operations outside the scope of (+-*/).
 
 
-## Training Details
+## Training
 
-### Training Data
-<!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-This model was trained on our Calculator-augmented set of
+This model was trained on [Calc-X](https://huggingface.co/collections/MU-NLPC/calc-x-652fee9a6b838fd820055483), a collection of math problem datasets which we converted into CoT with calculator interactions.
+We used a standard auto-regressive transformer training, i.e. a conditional next-token prediction with cross-entropy loss. For more detail about data, training or evaluation, see the [Calc-X and Calcformers paper](https://arxiv.org/abs/2305.15017).
 
-- [Calc Ape210k](https://huggingface.co/datasets/emnlp2023/Calc-ape210k) ([original Ape210k on github](https://github.com/Chenny0808/ape210k))
-- [Calc MathQA](https://huggingface.co/datasets/emnlp2023/Calc-math_qa) ([original MathQA on HF](https://huggingface.co/datasets/math_qa))
-- [Calc GSM8K](https://huggingface.co/datasets/emnlp2023/Calc-gsm8k) ([original GSM8K on HF](https://huggingface.co/datasets/gsm8k))
-- [Calc Aqua-RAT](https://huggingface.co/datasets/emnlp2023/Calc-aqua_rat) ([original Aqua-RAT on HF](https://huggingface.co/datasets/aqua_rat))
- 
-in a standard auto-regressive setup i.e. for a conditional next-token prediction with teacher-forced prefix.
 
 ## Cite