Update app.py

app.py

@@ -4,6 +4,7 @@ import cv2
 import shutil
 import sys
 from subprocess import call
+from utils import Restoration
 
 os.system("pip install dlib")
 os.system('bash setup.sh')
@@ -16,23 +17,25 @@ def run_cmd(command):
         sys.exit(1)
 
 def run(image):
-    os.makedirs("Temp")
-    os.makedirs("Temp/input")
-    print(type(image))
-    cv2.imwrite("Temp/input/input_img.png", 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)
+    # 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")
+    # result = cv2.imread("Temp/final_output/input_img.png")
+    # shutil.rmtree("Temp")
+
+    result = Restoration(image)
 
     return result
     
-iface = gr.Interface(fn=run, inputs="image", outputs="image").launch(debug=True)
+iface = gr.Interface(fn=run, inputs="image", outputs="image").launch(debug=True, share=True)