656

app.py

@@ -134,7 +134,7 @@ if 'image' in locals():
     if st.button('Enhance with SRFlow'):
         with st.spinner('Processing using SRFlow...'):
             with st.spinner('Wait for it... the model is processing the image'):
-                enhanced_image = return_SRFlow_result(image, divide=True)
+                enhanced_image = return_SRFlow_result(image)
                 st.session_state['srflow_enhanced_image'] = enhanced_image        
             st.session_state['srflow_clicked'] = True 
             st.success('Done!')

models/SRFlow/srflow.py

@@ -7,7 +7,7 @@ from PIL import Image
 import matplotlib.pyplot as plt
 from torchvision.transforms import PILToTensor, ToPILImage
 
-def return_SRFlow_result(lr, divide, conf_path='models/SRFlow/code/confs/SRFlow_DF2K_4X.yml'):
+def return_SRFlow_result(lr, divide=True, conf_path='models/SRFlow/code/confs/SRFlow_DF2K_4X.yml'):
     """
     Apply Super-Resolution using SRFlow model to the input LR (low-resolution) image.
 
@@ -44,7 +44,7 @@ def return_SRFlow_result(lr, divide, conf_path='models/SRFlow/code/confs/SRFlow_
     sr_img = Image.fromarray((sr).astype('uint8'))
     return sr_img
 
-def return_SRFlow_result_from_tensor(lr_tensor, divide=True):
+def return_SRFlow_result_from_tensor(lr_tensor, divide):
     """
     Apply Super-Resolution using SRFlow model to the input batched BCHW tensor.
 
@@ -68,7 +68,7 @@ def return_SRFlow_result_from_tensor(lr_tensor, divide=True):
 
     if not divide:
         sr_tensor = sr_tensor / 255
-        
+
     return sr_tensor 
 
 if __name__ == '__main__':