Update model.py

model.py

@@ -137,7 +137,7 @@ class Model:
         t1 = time.time()
         print(caption_embs.device)
         index_sample = generate(
-            self.gpt_model_canny,
+            self.gpt_model_edge,
             c_indices,
             (H // 16) * (W // 16),
             c_emb_masks,