Update anime_model.py

anime_model.py

@@ -17,7 +17,6 @@ from diffusers import (
 from settings import MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES
 
 class Model:
-    MODEL_ID = "lllyasviel/Annotators"
 
     def __init__(self):
         print("Initializing base model: ", base_model_id, " with ", task_name)
@@ -170,7 +169,7 @@ class Model:
         #if num_images > MAX_NUM_IMAGES:
         #    raise ValueError
         #self.load("NormalBae")
-        model = NormalBaeDetector.from_pretrained(self.MODEL_ID).to("cuda")
+        model = NormalBaeDetector.from_pretrained("lllyasviel/Annotators").to("cuda")
         torch.cuda.empty_cache()
         gc.collect()
         control_image = self.preprocess(