Update README.md

README.md

@@ -1,75 +0,0 @@
----
-license: gpl-3.0
-language:
-- en
----
-
-# ChartGemma: Visual Instruction-tuning for Chart Reasoning in the Wild
-
-
-Paper Link: https://arxiv.org/abs/2407.04172
-
-The abstract of the paper states that: 
-> Given the ubiquity of charts as a data analysis, visualization, and decision-making tool across industries and sciences, there has been a growing interest in developing pre-trained foundation models as well as general purpose instruction-tuned models for chart understanding and reasoning. However, existing methods suffer crucial drawbacks across two critical axes affecting the performance of chart representation models: they are trained on data generated from underlying data tables of the charts, ignoring the visual trends and patterns in chart images, \emph{and} use weakly aligned vision-language backbone models for domain-specific training, limiting their generalizability when encountering charts in the wild. We address these important drawbacks and introduce ChartGemma, a novel chart understanding and reasoning model developed over PaliGemma. Rather than relying on underlying data tables, ChartGemma is trained on instruction-tuning data generated directly from chart images, thus capturing both high-level trends and low-level visual information from a diverse set of charts. Our simple approach achieves state-of-the-art results across $5$ benchmarks spanning chart summarization, question answering, and fact-checking, and our elaborate qualitative studies on real-world charts show that ChartGemma generates more realistic and factually correct summaries compared to its contemporaries.
-# Web Demo
-If you wish to quickly try our model, you can access our public web demo hosted on the Hugging Face Spaces platform with a friendly interface!
-
-[ChartGemma Web Demo](https://huggingface.co/spaces/ahmed-masry/ChartGemma)
-
-# Inference
-You can easily use our models for inference with the huggingface library! 
-You just need to do the following:
-1. Chage the **_image_path_** to your chart example image path on your system
-2. Write the **_input_text_** 
-
-We recommend using beam search with a beam size of 4, but if your machine has low memory, you can remove the num_beams from the generate method. 
-```
-from PIL import Image
-import requests
-from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
-import torch
-
-torch.hub.download_url_to_file('https://raw.githubusercontent.com/vis-nlp/ChartQA/main/ChartQA%20Dataset/val/png/multi_col_1229.png', 'chart_example_1.png')
-
-image_path = "/content/chart_example_1.png"
-input_text ="program of thought: what is the sum of Faceboob Messnger and Whatsapp values in the 18-29 age group?"
-
-# Load Model
-model = PaliGemmaForConditionalGeneration.from_pretrained("ahmed-masry/chartgemma", torch_dtype=torch.float16)
-processor = AutoProcessor.from_pretrained("ahmed-masry/chartgemma")
-
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = model.to(device)
-
-# Process Inputs
-image = Image.open(image_path).convert('RGB')
-inputs = processor(text=input_text, images=image, return_tensors="pt")
-prompt_length = inputs['input_ids'].shape[1]
-inputs = {k: v.to(device) for k, v in inputs.items()}
-
-
-# Generate
-generate_ids = model.generate(**inputs, num_beams=4, max_new_tokens=512)
-output_text = processor.batch_decode(generate_ids[:, prompt_length:], skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
-print(output_text)
-
-```
-
-# Contact
-If you have any questions about this work, please contact **[Ahmed Masry](https://ahmedmasryku.github.io/)** using the following email addresses: **[email protected]** or **[email protected]**.
-
-# Reference
-Please cite our paper if you use our model in your research. 
-
-```
-@misc{masry2024chartgemmavisualinstructiontuningchart,
-      title={ChartGemma: Visual Instruction-tuning for Chart Reasoning in the Wild}, 
-      author={Ahmed Masry and Megh Thakkar and Aayush Bajaj and Aaryaman Kartha and Enamul Hoque and Shafiq Joty},
-      year={2024},
-      eprint={2407.04172},
-      archivePrefix={arXiv},
-      primaryClass={cs.AI},
-      url={https://arxiv.org/abs/2407.04172}, 
-}
-
-```