Update app.py

app.py

@@ -1,16 +1,16 @@
-import os
-import sys
-import argparse
-import time
-from pathlib import Path
-import pandas as pd
+# import os
+# import sys
+# import argparse
+# import time
+# from pathlib import Path
+# import pandas as pd
 
 import gradio as gr
-import cv2
+# import cv2
 from PIL import Image
-import torch
-import torch.backends.cudnn as cudnn
-from numpy import random
+# import torch
+# import torch.backends.cudnn as cudnn
+# from numpy import random
 import numpy as np
 
 # BASE_DIR = "/home/user/app"
@@ -212,9 +212,10 @@ def get_output(image):
     # if os.path.exists(output_path):
     #     return np.array(Image.open(output_path))
     # else:
-    return np.array(image)
+    return image
 
-gr.Interface(fn=get_output, 
-         inputs=gr.Image(type = "pil", label="Your image"),
-         outputs="image"
-         ).launch(debug=True)
+def main():
+    gr.Interface(fn=get_output, 
+             inputs=gr.Image(type = "pil", label="Your image"),
+             outputs="image"
+             ).launch(debug=True)