Update app.py

app.py

@@ -1,5 +1,3 @@
-import sys
-sys.path.append('SAFMN')
 
 import os
 import cv2
@@ -11,6 +9,7 @@ import torch
 import torch.nn.functional as F
 import gradio as gr
 
+from PIL import Image
 from utils.download_url import load_file_from_url
 from utils.color_fix import wavelet_reconstruction
 from models.safmn_arch import SAFMN
@@ -124,7 +123,7 @@ def inference(image, upscale, large_input_flag, color_fix):
 		upscale = 2
 
 	model = set_safmn(upscale)
-
+    
     img = np.array(image)
     img = img / 255.
     img = img.astype(np.float32)