Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 from flax.serialization import msgpack_restore
 
-msgpack_restore("PrakhAI/HelloWorld/checkpoint.msgpack")
+with open("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)