31

app.py

@@ -162,7 +162,7 @@ def sample_ddim(n_sample, n=20):
     return samples, intermediate
 
 # load in model weights and set to eval mode
-nn_model.load_state_dict(torch.load(f"{save_dir}/ft_context_model_4.pth", map_location=device))
+nn_model.load_state_dict(torch.load(f"{save_dir}/ft_context_model_31.pth", map_location=device))
 nn_model.eval()
 print("Loaded in Context Model")