Update app.py

app.py

@@ -12,6 +12,9 @@ from huggingface_hub import hf_hub_download
 
 from depth_anything_v2.dpt import DepthAnythingV2
 
+# Add this at the top to help with ZeroGPU authentication
+os.environ['SPACES_ZERO_GPU'] = '1'
+
 css = """
 #img-display-container {
     max-height: 100vh;
@@ -37,11 +40,11 @@ encoder2name = {
     'vits': 'Small',
     'vitb': 'Base',
     'vitl': 'Large',
-    'vitg': 'Giant', # we are undergoing company review procedures to release our giant model checkpoint
+    'vitg': 'Giant',
 }
 models = {}
 for encoder_key in model_configs.keys():
-    if encoder_key != 'vitg':  # Skip Giant model as mentioned it's under review
+    if encoder_key != 'vitg':
         try:
             model_name = encoder2name[encoder_key]
             model = DepthAnythingV2(**model_configs[encoder_key])
@@ -54,7 +57,6 @@ for encoder_key in model_configs.keys():
         except Exception as e:
             print(f"Failed to load {encoder2name[encoder_key]} model: {e}")
 
-# Set default model
 default_model = 'vitl' if 'vitl' in models else list(models.keys())[0]
 
 title = "# Depthinator"
@@ -68,7 +70,7 @@ def predict_depth(image, model_choice=None):
 
 def process_multiple_images(files, model_choice=None, include_bw=True):
     if not files:
-        return [], "No files uploaded"
+        return [], []
     
     if model_choice is None:
         model_choice = default_model
@@ -83,6 +85,8 @@ def process_multiple_images(files, model_choice=None, include_bw=True):
             
             if len(image_np.shape) == 3:
                 original_image = image_np.copy()
+                
+                # Call the GPU function for each image
                 depth = predict_depth(image_np[:, :, ::-1], model_choice)
                 
                 # Create colored depth map
@@ -127,6 +131,8 @@ def process_multiple_images(files, model_choice=None, include_bw=True):
                 
                 processed_files.extend([colored_depth_path.name, gray_depth_path.name, raw_depth_path.name])
                 
+                print(f"Successfully processed {base_name}")
+                
         except Exception as err:
             print(f"Error processing {file.name}: {str(err)}")
             continue
@@ -137,7 +143,6 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown(title)
     gr.Markdown(description)
     
-    # Create model choices for dropdown
     model_choices = [(f"{encoder2name[k]} ({k})", k) for k in models.keys()]
     
     with gr.Tabs():
@@ -183,7 +188,6 @@ with gr.Blocks(css=css) as demo:
         tmp_gray_depth = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
         gray_depth.save(tmp_gray_depth.name)
 
-        # Create black and white depth if requested
         tmp_bw_depth = None
         if include_bw:
             bw_depth = Image.fromarray(depth).convert('L')
@@ -196,4 +200,4 @@ with gr.Blocks(css=css) as demo:
     submit_multiple.click(process_multiple_images, inputs=[multiple_files, model_dropdown_multiple, include_bw_multiple], outputs=[gallery_output, download_files])
 
 if __name__ == '__main__':
-    demo.queue().launch(share=True)
+    demo.queue().launch()