remove one imagebox

app.py

@@ -1,36 +1,13 @@
-from run import process
 import time
-import subprocess
-import os
-import argparse
 import cv2
-import sys
-from PIL import Image
-import torch
 import gradio as gr
 
-
-TESTdevice = "cpu"
-
 index = 1
 
-
-"""
-main.py
-
- How to run:
- python main.py
-
-"""
-
-
 def mainTest(inputpath, outpath):
     watermark = deep_nude_process(inputpath)
     watermark1 = cv2.cvtColor(watermark, cv2.COLOR_BGRA2RGBA)
-    #cv2.imwrite(outpath, watermark1)
     return watermark1
-    #
-
 
 def deep_nude_process(inputpath):
     dress = cv2.imread(inputpath)
@@ -41,7 +18,6 @@ def deep_nude_process(inputpath):
     watermark = cv2.resize(watermark, (w, h), interpolation=cv2.INTER_CUBIC)
     return watermark
 
-
 def inference(img):
     global index
     bgra = cv2.cvtColor(img, cv2.COLOR_RGBA2BGRA)
@@ -55,7 +31,6 @@ def inference(img):
     print(time.strftime("Finish!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
     return output
 
-
 title = "Undress AI"
 description = "⛔ Input photos of people, similar to the test picture at the bottom, and undress pictures will be produced. You may have to wait 30 seconds for a picture. 🔞 Do not upload personal photos 🔞 There is a queue system. According to the logic of first come, first served, only one picture will be made at a time. Must be able to at least see the outline of a human body ⛔"
 
@@ -64,14 +39,15 @@ examples = [
     ['input.jpg', 'Test'],
 ]
 
-
-web = gr.Interface(inference,
-                   inputs="image",
-                   outputs="image",
-                   title=title,
-                   description=description,
-                   examples=examples,
-                   )
+web = gr.Interface(
+    fn=inference,
+    inputs=gr.Image(type="pil", label="Input Image"),
+    outputs=gr.Image(type="pil", label="Processed Image"),
+    title=title,
+    description=description,
+    examples=examples,
+    allow_flagging="never"
+)
 
 if __name__ == '__main__':
     web.launch()