From c6fcf6c438bf5c447c6d3c4dfd0b36c36aa1e4e3 Mon Sep 17 00:00:00 2001
From: AlUlkesh <99896447+AlUlkesh@users.noreply.github.com>
Date: Sun, 16 Jul 2023 09:15:29 +0200
Subject: [PATCH] ImageReward fix, #214

---
 install.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/install.py b/install.py
index d87ff17..60fa80f 100644
--- a/install.py
+++ b/install.py
@@ -6,12 +6,13 @@ if not launch.is_installed("send2trash"):
 
 try:
     import ImageReward
-    import dataset
+    import datasets
     import dill
     import diffusers
     import multiprocessing
     import pyarrow
     import xxhash
-except ImportError as e:
+except (ImportError, ModuleNotFoundError) as e:
+    #print(e)
     req_IR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "req_IR.txt")
     launch.run_pip(f'install -r "{req_IR}" --no-deps image-reward', 'ImageReward requirement for image browser')