Gradio 6.29; local BiRefNet-lite via rembg (remote ysharma Space is dead); fonts via packages.txt; lazy example cache

README.md

@@ -4,7 +4,8 @@ emoji: 📈
 colorFrom: gray
 colorTo: yellow
 sdk: gradio
-sdk_version: 5.5.0
+sdk_version: 6.29.0
+python_version: "3.12"
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -3,10 +3,11 @@ from PIL import Image, ImageDraw, ImageFont
 import numpy as np
 import os
 from PIL import Image
-from gradio_client import Client, handle_file
-import uuid
+from rembg import new_session, remove
 
-client = Client("ysharma/BiRefNet_for_text_writing")
+# The original called the Space ysharma/BiRefNet_for_text_writing, which no longer builds.
+# Same model family (BiRefNet lite), run locally on CPU through rembg's ONNX session.
+session = new_session("birefnet-general-lite")
 
 def add_text_with_stroke(draw, text, x, y, font, text_color, stroke_width):
     """Helper function to draw text with stroke"""
@@ -16,12 +17,7 @@ def add_text_with_stroke(draw, text, x, y, font, text_color, stroke_width):
             draw.text((x + adj_x, y + adj_y), text, font=font, fill=text_color)
 
 def remove_background(image):
-    # Save the image to a specific location
-    filename = f"image_{uuid.uuid4()}.png"  # Generates a universally unique identifier (UUID) for the filename
-    image.save(filename)
-    # Call gradio client for background removal
-    result = client.predict(images=handle_file(filename), api_name="/image")
-    return Image.open(result[0])
+    return remove(image.convert("RGB"), session=session)
 
 def superimpose(image_with_text, overlay_image):
     # Open image as RGBA to handle transparency
@@ -49,7 +45,7 @@ def add_text_to_image(
     if input_image is None:
         return None
     
-    image = Image.fromarray(input_image)
+    image = Image.fromarray(input_image).convert("RGB")
     # remove background 
     overlay_image = remove_background(image)
 
@@ -214,21 +210,14 @@ def create_interface():
             outputs=output_image,
             fn=add_text_to_image,
             cache_examples=True,
+            cache_mode="lazy",
         )
 
     return app
 
 # Launch the app
 if __name__ == "__main__":
-    # Try to install required font
-    try:
-        import subprocess
-        subprocess.run(['apt-get', 'update'])
-        subprocess.run(['apt-get', 'install', '-y', 'fonts-dejavu'])
-        print("Font installed successfully")
-    except:
-        print("Could not install font automatically. Please install DejaVu font manually.")
-    
+    # DejaVu fonts come from packages.txt (fonts-dejavu-core); runtime apt-get can't run as non-root.
     # Create and launch the interface
     app = create_interface()
     app.launch()

packages.txt

@@ -0,0 +1 @@
+fonts-dejavu-core

requirements.txt

@@ -0,0 +1,3 @@
+rembg[cpu]
+pillow
+numpy