Update app/app_savta.py

app/app_savta.py

@@ -4,14 +4,15 @@ from pathlib import Path
 import torch
 from fastai.vision.all import *
 import gradio as gr
+from gradio.flagging import HuggingFaceDatasetSaver
 
 ######################
 # Hugging Face flags #
 ######################
 
 HF_TOKEN = os.getenv("HF_TOKEN")
-# `HuggingFaceDatasetSaver` is still available in Gradio ≥ 5
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "savtadepth-flags-V2")
+# Use the new import path for Gradio ≥ 5
+hf_writer = HuggingFaceDatasetSaver(repo_id="savtadepth-flags-V2", token=HF_TOKEN)
 
 ############
 #   DVC    #
@@ -59,7 +60,7 @@ class ImageImageDataLoaders(DataLoaders):
 
 
 def get_y_fn(x: Path) -> Path:
-    """Map an RGB image path to its depth‑map counterpart."""
+    """Map an RGB image path to its depth-map counterpart."""
     return Path(str(x).replace(".jpg", "_depth.png"))
 
 
@@ -92,7 +93,7 @@ learner.load("model")
 #####################
 
 def predict_depth(input_img: PILImage) -> PILImageBW:
-    """Generate a single‑channel depth prediction from an RGB image."""
+    """Generate a single-channel depth prediction from an RGB image."""
     depth, *_ = learner.predict(input_img)
     return PILImageBW.create(depth).convert("L")
 
@@ -105,7 +106,7 @@ title = "SavtaDepth WebApp"
 description = (
     """
     <p style="text-align:center;">
-        Savta Depth is a collaborative Open‑Source project for monocular depth estimation – turn 2‑D photos into 3‑D. 🏞️<br>
+        Savta Depth is a collaborative Open-Source project for monocular depth estimation – turn 2‑D photos into 3‑D.<br>
         Try the model below or explore the resources.
         <br><img src="https://huggingface.co/spaces/kingabzpro/savtadepth/resolve/main/examples/cover.png" width="250"/>
     </p>
@@ -116,7 +117,7 @@ article = (
     """
     <p style='text-align:center'>
         <a href='https://dagshub.com/OperationSavta/SavtaDepth' target='_blank'>Project on DAGsHub</a> •
-        <a href='https://colab.research.google.com/drive/1XU4DgQ217_hUMU1dllppeQNw3pTRlHy1?usp=sharing' target='_blank'>Google Colab Demo</a>
+        <a href='https://colab.research.google.com/drive/1XU4DgQ217_hUMU1dllppeQNw3pTRlHy1?usp=sharing' target='_blank'>Google Colab Demo</a>
         <br/>
         <img src='https://visitor-badge.glitch.me/badge?page_id=kingabzpro/savtadepth' alt='visitor badge'/>
     </p>
@@ -144,7 +145,7 @@ with gr.Blocks(title=title) as demo:
         flagging_options=["incorrect", "worst", "ambiguous"],
         flagging_callback=hf_writer,
         examples=examples,
-        cache_examples=False,  # use live inference for examples
+        cache_examples=False,
         theme=gr.themes.Soft(),
     )