Update app.py

app.py

@@ -7,7 +7,7 @@ from PIL import Image
 from scipy.interpolate import Rbf
 
 # Load MiDaS depth estimation model
-midas_model = torch.hub.load("intel-isl/MiDaS", "MiDaS_small")
+midas_model = torch.hub.load("intel-isl/MiDaS", "DPT_Hybrid")
 midas_model.eval()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 midas_model.to(device)
@@ -73,6 +73,9 @@ def blend_design(cloth_img, design_img):
     return Image.fromarray(cloth_np)
 
 def main(cloth, design):
+    global midas_model
+    if midas_model is None:
+        midas_model = torch.hub.load("intel-isl/MiDaS", "MiDaS_small").to(device).eval()
     return blend_design(cloth, design)
 
 iface = gr.Interface(
@@ -84,4 +87,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(share=True, debug=True, queue=True)