app.py

import streamlit as st
from huggingface_hub import HfFileSystem
from flax.serialization import msgpack_restore
import os

fs = HfFileSystem()

print(fs.glob("/PrakhAI"))

with open("models/PrakhAI/HelloWorld/checkpoint.msgpack", "rb") as f:
    model = msgpack_restore(f.read())

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