first commit in hf_spaces

deoldify/visualize.py

@@ -8,6 +8,7 @@ from PIL import Image
 from matplotlib.axes import Axes
 from matplotlib.figure import Figure
 from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
+from typing import Tuple
 
 import torch
 from fastai.core import *
@@ -18,6 +19,7 @@ from .generators import gen_inference_deep, gen_inference_wide
 
 
 
+# class LoadedModel
 class ModelImageVisualizer:
     def __init__(self, filter: IFilter, results_dir: str = None):
         self.filter = filter
@@ -29,11 +31,11 @@ class ModelImageVisualizer:
         # gc.collect()
 
     def _open_pil_image(self, path: Path) -> Image:
-        return PIL.Image.open(path).convert('RGB')
+        return Image.open(path).convert('RGB')
 
     def _get_image_from_url(self, url: str) -> Image:
         response = requests.get(url, timeout=30, headers={'Accept': '*/*;q=0.8'})
-        img = PIL.Image.open(BytesIO(response.content)).convert('RGB')
+        img = Image.open(BytesIO(response.content)).convert('RGB')
         return img
 
     def plot_transformed_image_from_url(
@@ -41,7 +43,7 @@ class ModelImageVisualizer:
         url: str,
         path: str = 'test_images/image.png',
         results_dir:Path = None,
-        figsize: (int, int) = (20, 20),
+        figsize: Tuple[int, int] = (20, 20),
         render_factor: int = None,
         
         display_render_factor: bool = False,
@@ -66,7 +68,7 @@ class ModelImageVisualizer:
         self,
         path: str,
         results_dir:Path = None,
-        figsize: (int, int) = (20, 20),
+        figsize: Tuple[int, int] = (20, 20),
         render_factor: int = None,
         display_render_factor: bool = False,
         compare: bool = False,
@@ -95,7 +97,7 @@ class ModelImageVisualizer:
     def plot_transformed_pil_image(
         self,
         input_image: Image,
-        figsize: (int, int) = (20, 20),
+        figsize: Tuple[int, int] = (20, 20),
         render_factor: int = None,
         display_render_factor: bool = False,
         compare: bool = False,
@@ -117,7 +119,7 @@ class ModelImageVisualizer:
 
     def _plot_comparison(
         self,
-        figsize: (int, int),
+        figsize: Tuple[int, int],
         render_factor: int,
         display_render_factor: bool,
         orig: Image,
@@ -141,7 +143,7 @@ class ModelImageVisualizer:
 
     def _plot_solo(
         self,
-        figsize: (int, int),
+        figsize: Tuple[int, int],
         render_factor: int,
         display_render_factor: bool,
         result: Image,
@@ -172,9 +174,6 @@ class ModelImageVisualizer:
             orig_image, orig_image, render_factor=render_factor,post_process=post_process
         )
 
-        # if watermarked:
-        #     return get_watermarked(filtered_image)
-
         return filtered_image
 
     def get_transformed_pil_image(
@@ -208,7 +207,7 @@ class ModelImageVisualizer:
                 backgroundcolor='black',
             )
 
-    def _get_num_rows_columns(self, num_images: int, max_columns: int) -> (int, int):
+    def _get_num_rows_columns(self, num_images: int, max_columns: int) -> Tuple[int, int]:
         columns = min(num_images, max_columns)
         rows = num_images // columns
         rows = rows if rows * columns == num_images else rows + 1