Upload app.py

app.py

@@ -70,7 +70,7 @@ def main():
     # small_model = load_casey_model(tokenizer, device)
     model = load_big_model(tokenizer, device)
 
-    st.markdown(print(model))
+    st.markdown([p.numel() for p in model.parameters()])
 
     if st.button('Write my story!'):
         placeholder = st.empty()