AAhad's picture
added basic models
a19213a
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)