Added background options, bug fixes

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+Backgrounds/3.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,46 +1,33 @@
-import torch
 import gradio as gr
-from gradio import processing_utils, utils
+import torch
+
 from gradio_depth_pred import create_demo as create_depth_pred_demo
-from PIL import Image
-import random
-from diffusers import (
-    DiffusionPipeline,
-    AutoencoderKL,
-    StableDiffusionControlNetPipeline,
-    ControlNetModel,
-    StableDiffusionLatentUpscalePipeline,
-    StableDiffusionImg2ImgPipeline,
-    StableDiffusionControlNetImg2ImgPipeline,
-    DPMSolverMultistepScheduler,  # <-- Added import
-    EulerDiscreteScheduler  # <-- Added import
-)
 
-print(f"Is CUDA available: {torch.cuda.is_available()}")
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+css = """
+#img-display-container {
+    max-height: 50vh;
+    }
+#img-display-input {
+    max-height: 40vh;
+    }
+#img-display-output {
+    max-height: 40vh;
+    }
+    
+"""
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
+model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to(DEVICE).eval()
 
-import time
-from style import css
-
-BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
+title = "# Ultra Heroes"
+description = """Official demo for **Ultra Heroes**.]
 
-title = "Ultra Heroes"
-description = "Testing composites and lighting tweaks."
+with gr.Blocks(css=css) as demo:
+    gr.Markdown(title)
+    gr.Markdown(description)
+    with gr.Tab("Depth Prediction"):
+        create_depth_pred_demo(model)
 
-def inference(text):
-  output_flan = ""
-  output_vanilla = ""
-  return [output_flan, output_vanilla]
+    # gr.HTML('''<br><br><br><center>You can duplicate this Space to skip the queue:<a href="https://huggingface.co/spaces/shariqfarooq/ZoeDepth?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a></center>''')
 
-io = gr.Interface(
-  inference,
-  gr.Textbox(lines=3),
-  outputs=[
-    gr.Textbox(lines=3, label="Flan T5"),
-    gr.Textbox(lines=3, label="T5")
-  ],
-  title=title,
-  description=description,
-)
-io.launch()
+if __name__ == '__main__':
+    demo.queue().launch()