Create app.py

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+import os
+import cv2 
+import shutil
+import sys
+from subprocess import call
+
+os.system('bash setup.sh')
+
+def run_cmd(command):
+    try:
+        call(command, shell=True)
+    except KeyboardInterrupt:
+        print("Process interrupted")
+        sys.exit(1)
+
+def run(image):
+    os.makedirs("Temp")
+    os.makedirs("Temp/input")
+    print(type(image))
+    cv2.imwrite("Temp/input/input_img.png", image)
+
+    command = ("python run.py --input_folder "
+            + "Temp/input"
+            + " --output_folder "
+            + "Temp"
+            + " --GPU "
+            + "-1"
+            + " --with_scratch")
+    run_cmd(command)
+
+    result = cv2.imread("Temp/final_output/input_img.png")
+    shutil.rmtree("Temp")
+
+    return result
+    
+iface = gr.Interface(fn=run, inputs="image", outputs="image").launch(debug=True)