Sasikanth commited on
Commit
57a50dc
1 Parent(s): cacc6ce
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,6 +27,7 @@ def select_model():
27
  MODEL_NAME = 'diet_nerf_ship'
28
  return MODEL_DIR,MODEL_NAME,FILE_ID
29
 
 
30
 
31
  @st.cache
32
  def download_model():
@@ -50,7 +51,6 @@ model_path = os.path.join(MODEL_DIR, MODEL_NAME)
50
  if not os.path.isfile(model_path):
51
  download_model()
52
 
53
- MODEL_DIR,MODEL_NAME,FILE_ID = select_model()
54
  model, state = fetch_model()
55
  pi = math.pi
56
  st.sidebar.image("images/diet-nerf.png", width=310)
 
27
  MODEL_NAME = 'diet_nerf_ship'
28
  return MODEL_DIR,MODEL_NAME,FILE_ID
29
 
30
+ MODEL_DIR,MODEL_NAME,FILE_ID = select_model()
31
 
32
  @st.cache
33
  def download_model():
 
51
  if not os.path.isfile(model_path):
52
  download_model()
53
 
 
54
  model, state = fetch_model()
55
  pi = math.pi
56
  st.sidebar.image("images/diet-nerf.png", width=310)