import streamlit as st
import streamlit as st
import pandas as pd
from st_aggrid import AgGrid, GridOptionsBuilder
import numpy as np

st.title('List of LLM Models 🔥')

st.warning("WIP: There are many more models to be added.")

data = {
  "Task": [ 'Text Classification', 'Text Generation', 'Summarization', 'Image Classification',
		   'Image Segmentation', 'Object Detection', 'Audio Classification', 'Automatic Speech Recognition' ,'Visual question answering',
            'Document question answering', 'Image captioning' ],
  "Description": [ 'assign a label to a given sequence of text', 'generate text given a prompt', 'generate a summary of a sequence of text or document', 'assign a label to an image',
				  'assign a label to each individual pixel of an image (supports semantic, panoptic, and instance segmentation)', 'predict the bounding boxes and classes of objects in an image',
				   'assign a label to some audio data' ,'transcribe speech into text', 'answer a question about the image, given an image and a question',
					'answer a question about the document, given a document and a question',  'generate a caption for a given image'  ],
  "Modality": [ 'NLP','NLP','NLP','Computer Vision','Computer Vision','Computer Vision', 'Audio','Audio', 'Multimodal', 'Multimodal', 'Multimodal']
}

df1 = pd.DataFrame(data)
my_table = st.table(df1)