import streamlit as st
from modeling_hybrid_clip import FlaxHybridCLIP


@st.cache
def get_model():
    return FlaxHybridCLIP.from_pretrained("clip-italian/clip-italian")


"""
# CLIP Italian Demo (Flax Community Week)
"""

x = st.slider("Select a value")
st.write(x, "squared is", x * x)

model = get_model()
st.write(str(model.config["architectures"]))