Update app.py

app.py

@@ -1,10 +1,9 @@
 import gradio as gr
 import os
-#import cv2 
-#import shutil
-#import sys
-#from subprocess import call
-from utils import Restoration
+import cv2 
+import shutil
+import sys
+from subprocess import call
 
 os.system("pip install dlib")
 os.system('bash setup.sh')
@@ -17,25 +16,22 @@ 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)
-
-    # result = cv2.imread("Temp/final_output/input_img.png")
-    # shutil.rmtree("Temp")
-
-    result = Restoration(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, share=True)