Update app.py

app.py

@@ -99,4 +99,4 @@ 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"This is how the model architecture: {model.hf_device_map}")
+        st.write(f"The following is the distribution of model (in case n GPUs > 1): {model.hf_device_map}")