Update app.py

app.py

@@ -53,7 +53,7 @@ def load_img2mesh_model(model_name):
     return i2m_model, base_diffusion_i2m
 
 img2mesh_model_name = 'base40M' #'base300M' #'base1B'
-img2mesh_model, base_diffusion_i2m = load_img2mesh_model(img2mesh_model_name)
+i2m_model, base_diffusion_i2m = load_img2mesh_model(img2mesh_model_name)
 
 
 set_state('Creating upsample model...')
@@ -78,14 +78,14 @@ def get_sampler(model_name, txt2obj, guidance_scale):
 
     global img2mesh_model_name
     global base_diffusion_i2m
-    global img2mesh_model
+    global i2m_model
     if model_name != img2mesh_model_name:
         img2mesh_model_name = model_name
-        img2mesh_model, base_diffusion_i2m = load_img2mesh_model(model_name)
+        i2m_model, base_diffusion_i2m = load_img2mesh_model(model_name)
 
     return PointCloudSampler(
             device=device,
-            models=[t2m_model, upsampler_model],
+            models=[t2m_model if txt2obj else i2m_model, upsampler_model],
             diffusions=[base_diffusion_t2m if txt2obj else base_diffusion_i2m, upsampler_diffusion],
             num_points=[1024, 4096 - 1024],
             aux_channels=['R', 'G', 'B'],