Update app.py

app.py

@@ -1,6 +1,6 @@
+# Imports
 import torch
 from tqdm.auto import tqdm
-
 from point_e.diffusion.configs import DIFFUSION_CONFIGS, diffusion_from_config
 from point_e.diffusion.sampler import PointCloudSampler
 from point_e.models.download import load_checkpoint
@@ -8,7 +8,7 @@ from point_e.models.configs import MODEL_CONFIGS, model_from_config
 from point_e.util.plotting import plot_point_cloud
 import streamlit as st
 
-
+# Downloads
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 st.write('creating base model...')
 base_name = 'base40M-textvec'
@@ -24,7 +24,6 @@ base_model.load_state_dict(load_checkpoint(base_name, device))
 st.write('downloading upsampler checkpoint...')
 upsampler_model.load_state_dict(load_checkpoint('upsample', device))
 
-
 # Define Sampler
 sampler = PointCloudSampler(
     device=device,