samusander commited on
Commit
b01cec1
1 Parent(s): c041925

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,6 +1,6 @@
 
1
  import torch
2
  from tqdm.auto import tqdm
3
-
4
  from point_e.diffusion.configs import DIFFUSION_CONFIGS, diffusion_from_config
5
  from point_e.diffusion.sampler import PointCloudSampler
6
  from point_e.models.download import load_checkpoint
@@ -8,7 +8,7 @@ from point_e.models.configs import MODEL_CONFIGS, model_from_config
8
  from point_e.util.plotting import plot_point_cloud
9
  import streamlit as st
10
 
11
-
12
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
13
  st.write('creating base model...')
14
  base_name = 'base40M-textvec'
@@ -24,7 +24,6 @@ base_model.load_state_dict(load_checkpoint(base_name, device))
24
  st.write('downloading upsampler checkpoint...')
25
  upsampler_model.load_state_dict(load_checkpoint('upsample', device))
26
 
27
-
28
  # Define Sampler
29
  sampler = PointCloudSampler(
30
  device=device,
 
1
+ # Imports
2
  import torch
3
  from tqdm.auto import tqdm
 
4
  from point_e.diffusion.configs import DIFFUSION_CONFIGS, diffusion_from_config
5
  from point_e.diffusion.sampler import PointCloudSampler
6
  from point_e.models.download import load_checkpoint
 
8
  from point_e.util.plotting import plot_point_cloud
9
  import streamlit as st
10
 
11
+ # Downloads
12
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
13
  st.write('creating base model...')
14
  base_name = 'base40M-textvec'
 
24
  st.write('downloading upsampler checkpoint...')
25
  upsampler_model.load_state_dict(load_checkpoint('upsample', device))
26
 
 
27
  # Define Sampler
28
  sampler = PointCloudSampler(
29
  device=device,