imports cleaned up

predict.py

@@ -4,14 +4,12 @@
 from types import SimpleNamespace
 from typing import List
 import subprocess
-import argparse
 import torch
 import torchvision.transforms.functional as TF
-import models
 import os
 import cv2
 from tqdm import tqdm
-from utility import tensor2rgb, FoldUnfold
+from utility import tensor2rgb
 from cog import BasePredictor, Input, Path
 
 import torch