Update app.py

app.py

@@ -17,7 +17,7 @@ import gradio as gr
 # from moge.model.v1 import MoGeModel
 # from utils.geometry import compute_pointmap
 # import os, shutil
-# import cv2
+import cv2
 # from huggingface_hub import hf_hub_download
 # from PIL import Image
 # import matplotlib.pyplot as plt
@@ -196,9 +196,22 @@ import gradio as gr
 #     rgb = output_np[:,:,:3]
 #     return mask, rgb 
 
+@GPU(duration = 180)
 def process_image(input_img):
+
+    rotated = False
+    if input_img.shape[0] > input_img.shape[1]:
+        input_img = cv2.rotate(input_img, cv2.ROTATE_90_COUNTERCLOCKWISE)
+        rotated = True
+    input_img = cv2.resize(input_img, (640, 480))
+    mask, rgb = mask_rembg(input_img)
+    depth_dict = depth_moge(input_img)
+    if os.path.exists(outdir):
+        shutil.rmtree(outdir)
+    os.makedirs(outdir)
+    
     return input_img, input_img
-# @GPU(duration = 180)
+
 # def process_image(input_img):
 #     # resize the input image
 #     rotated = False