Update colorization.py

colorization.py

@@ -20,7 +20,7 @@ def colorize_image(image):
     return f'./output/DeOldify/'+Path(image.name).stem+".png"
 
 # def inference(img, version, scale, weight):
-def inferenceColorize(img, version, scale):
+def inferenceColorize(img, scale):
     # weight /= 100
     print(img, scale)
     '''
@@ -56,21 +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 img, save_path
     '''
     except Exception as error:
         print('global exception', error)
         return None, None
-'''
+    '''
 
 colorize = gr.Interface(
     inferenceColorize, [
@@ -91,7 +91,6 @@ colorize = gr.Interface(
     
 colorize.queue(concurrency_count=4)
 colorize.launch()
-
 '''
 def create_interface():
     with gr.Blocks() as enhancer: