Update vtoonify_model.py

vtoonify_model.py

@@ -211,7 +211,9 @@ class Model():
             print('*** Toonify %dx%d image with style of %s'%(y_tilde.shape[2], y_tilde.shape[3], style_type))
         
         return ((y_tilde[0].cpu().numpy().transpose(1, 2, 0) + 1.0) * 127.5).astype(np.uint8), 'Successfully toonify the image with style of %s' % (self.style_name)
-
+    except Exception as e:
+        logging.error(f"Error during model execution: {e}")
+        return np.zeros((256, 256, 3), np.uint8), f"Error during processing: {str(e)}"
     def tensor2cv2(self, img):
         """Convert a tensor image to OpenCV format."""
         tmp = ((img.cpu().numpy().transpose(1, 2, 0) + 1.0) * 127.5).astype(np.uint8).copy()