Update vtoonify_model.py

vtoonify_model.py

@@ -195,7 +195,7 @@ class Model():
             return np.zeros((256, 256, 3), np.uint8), 'Oops, something wrong with the style type. Please go to Step 1 and load model again.'
             
         try:
-           with torch.no_grad():
+            with torch.no_grad():
                if self.color_transfer:
                    s_w = exstyle
                else:
@@ -212,9 +212,11 @@ class Model():
                logging.info(f"Output from VToonify shape: {y_tilde.shape}")
                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)
-    
- 
+            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()