cleanup

app.py

@@ -1,28 +1,29 @@
-from torch.utils.data import DataLoader
-import torch
-from model.base.geometry import Geometry
-from common.evaluation import Evaluator
-from common.logger import AverageMeter
-from common.logger import Logger
-from data import download
-from model import chmnet
+import os
+import random
 from itertools import product
+
+import gdown
+import gradio as gr
 import matplotlib
 import matplotlib.patches as patches
-from matplotlib.patches import ConnectionPatch
-from matplotlib import pyplot as plt
-from PIL import Image
 import numpy as np
-import os
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
 import torchvision
+import torchvision.models as models
 import torchvision.transforms as transforms
 import torchvision.transforms.functional as TF
-import torchvision.models as models
-import torch.nn as nn
-import torch.nn.functional as F
-import random
-import gradio as gr
-import gdown
+from matplotlib import pyplot as plt
+from matplotlib.patches import ConnectionPatch
+from PIL import Image
+from torch.utils.data import DataLoader
+
+from common.evaluation import Evaluator
+from common.logger import AverageMeter, Logger
+from data import download
+from model import chmnet
+from model.base.geometry import Geometry
 
 # Downloading the Model
 gdown.download(id="1zsJRlAsoOn5F0GTCprSFYwDDfV85xDy6", output="pas_psi.pt", quiet=False)