From d436e82f9fc03af1758b00a230194173a49674db Mon Sep 17 00:00:00 2001 From: ProbablyButter Date: Mon, 25 Nov 2024 18:51:15 -0800 Subject: [PATCH] added zookie resources --- cog_modules/random/cog.py | 2 +- resources.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) 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()