app update

app.py

@@ -70,6 +70,8 @@ def main():
     # small_model = load_casey_model(tokenizer, device)
     model = load_big_model(tokenizer, device)
 
+    st.markdown(f"Model Type: {type(model)}, Device: {device}")
+
     if st.button('Write my story!'):
         placeholder = st.empty()
         # if model_version == 'smoll':