diff --git a/cog_modules/random/cog.py b/cog_modules/random/cog.py index 781489e..0ff4d14 100644 --- a/cog_modules/random/cog.py +++ b/cog_modules/random/cog.py @@ -10,7 +10,7 @@ from dotenv import load_dotenv from gizmopics import * -from resources import gizmoResources, taoResources +from resources import gizmoResources, taoResources, zookieResources load_dotenv() CATS = os.getenv("x-api-key") diff --git a/resources.py b/resources.py index b424012..06b3dda 100644 --- a/resources.py +++ b/resources.py @@ -24,3 +24,9 @@ taoResources = pickle.load(f) except FileNotFoundError: taoResources = Counter() + +try: + with open("zookieResources", "rb") as f: + zookieResources = pickle.load(f) +except FileNotFoundError: + zookieResources = Counter()