Upload app.py

app.py

@@ -70,7 +70,9 @@ def main():
     # small_model = load_casey_model(tokenizer, device)
     model = load_big_model(tokenizer, device)
 
-    st.markdown([p.numel() for p in model.parameters()])
+    st.markdown("Figure it out you broken piece of shit, fucks sake")
+    st.markdown(model.state_dict())
+
 
     if st.button('Write my story!'):
         placeholder = st.empty()