webplip / app.py
vinid's picture
Upload 4 files
61448a4
raw
history blame
600 Bytes
import streamlit as st
import pandas as pd
import home
import numpy as np
from PIL import Image
import requests
import transformers
import text2image
import tokenizers
from io import BytesIO
import streamlit as st
from transformers import CLIPModel
import clip
import torch
from transformers import (
VisionTextDualEncoderModel,
AutoFeatureExtractor,
AutoTokenizer
)
from transformers import AutoProcessor
st.sidebar.title("Explore our PLIP Demo")
PAGES = {
"Introduction": home,
"Text to Image": text2image,
}
page = st.sidebar.radio("", list(PAGES.keys()))
PAGES[page].app()