updated app

app.py

@@ -2,10 +2,12 @@ import os, sys
 import argparse
 import numpy as np
 import torch
-from torchvision import transforms
 import matplotlib.pyplot as plt
 from PIL import Image
 
+from ROME.src.utils import args as args_utils
+from ROME.src.utils.processing import process_black_shape, tensor2image
+
 # loading models ---- create model repo
 from huggingface_hub import hf_hub_url
 default_modnet_path = hf_hub_url('Pie31415/rome','modnet_photographic_portrait_matting.ckpt')