Update app.py

app.py

@@ -98,4 +98,5 @@ if model_loaded:
 
         # Display GPU memory information after generation
         gpu_memory_after_generation = get_gpu_memory()
-        st.write(f"GPU Memory Info after generation: {gpu_memory_after_generation}")
+        st.write(f"GPU Memory Info after generation: {gpu_memory_after_generation}")
+        st.write(f"This is how the model architecture: {model.hf_device_map}")