update space

app.py

@@ -170,12 +170,12 @@ def main():
         ckp = torch.load(hf_hub_download(repo_id='silentchen/Shap_Editor', subfolder='single', filename='{}.pt'.format(editing_type)), map_location='cpu')
         tmp_model.load_state_dict(ckp['model'])
         tmp_model.eval()
-        # print("loaded latent model")
+        print("loaded latent model")
         tmp_model.to(device)
         noise_initial = ckp['initial_noise']['noise'].to(device)
         initial_noise[editing_type] = noise_initial
         noise_start_t[editing_type] = ckp['t_start']
-        models[editing_type] = tmp_model.to(device)
+        models[editing_type] = tmp_model
     @torch.no_grad()
     def optimize_all(prompt, instruction,
                      rand_seed):