webplip / app.py
vinid's picture
twitter iframe (#3)
dc3cb2a
raw
history blame
650 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 zeroshot
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,
"Image Prediction": zeroshot,
}
page = st.sidebar.radio("", list(PAGES.keys()))
PAGES[page].app()