Update anime_model.py

anime_model.py

@@ -169,8 +169,8 @@ class Model:
         #    raise ValueError
         #if num_images > MAX_NUM_IMAGES:
         #    raise ValueError
-        self.load("NormalBae")
-        model = NormalBaeDetector.from_pretrained(self.MODEL_ID)#.to("cuda")
+        #self.load("NormalBae")
+        model = NormalBaeDetector.from_pretrained(self.MODEL_ID).to("cuda")
         torch.cuda.empty_cache()
         gc.collect()
         control_image = self.preprocess(