Update colorization.py

colorization.py

@@ -8,7 +8,7 @@ from typing import Optional
 
 model = hub.Module(name='deoldify')
 # NOTE:  Max is 45 with 11GB video cards. 35 is a good default
-render_factor=35
+render_factor=40
 
 def colorize_image(image):
     # now = datetime.now().strftime("%Y%m%d-%H%M%S-%f")
@@ -23,6 +23,7 @@ def colorize_image(image):
 def inferenceColorize(img, version, scale):
     # weight /= 100
     print(img, scale)
+    '''
     try:
         extension = os.path.splitext(os.path.basename(str(img)))[1]
         img = cv2.imread(img, cv2.IMREAD_UNCHANGED)
@@ -55,17 +56,21 @@ def inferenceColorize(img, version, scale):
             extension = 'png'
         else:
             extension = 'jpg'
-        
+        '''
+        os.makedirs('output', exist_ok=True)
         model.predict(img.name)
         save_path = f'output/DeOldify/'+Path(img.name).stem+'.{extension}'
     
+        '''
         cv2.imwrite(save_path, output)
         output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
-        return output, save_path
+        '''
+        return img, save_path
+    '''
     except Exception as error:
         print('global exception', error)
         return None, None
-
+'''
 
 colorize = gr.Interface(
     inferenceColorize, [