test

app.py

@@ -14,6 +14,8 @@ from transformers import AutoModelForImageSegmentation
 from torchvision import transforms
 import torch
 import subprocess
+import base64
+
 subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 generated_images = []
 
@@ -35,6 +37,17 @@ def remove_bg(image):
     mask = transforms.ToPILImage()(preds).resize(im.size)
     return mask
 
+
+
+
+def get_base64_logo(path="examples/logo.png"):
+    image = Image.open(path).convert("RGBA")
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    base64_img = base64.b64encode(buffered.getvalue()).decode()
+    return f"data:image/png;base64,{base64_img}"
+
+
 class DreamFuseGUI:
     def __init__(self):
         self.examples = [
@@ -440,6 +453,7 @@ class DreamFuseGUI:
         ), draggable_img
 
 
+
     def create_gui(self):
         config = InferenceConfig()
         config.lora_id = 'LL3RD/DreamFuse'
@@ -450,10 +464,11 @@ class DreamFuseGUI:
 
         with gr.Blocks(css=self.css_style) as demo:
             modified_fg_state = gr.State() 
+            logo_data_url = get_base64_logo()
             gr.HTML(
-                """
+                f"""
                 <div style="display: flex; align-items: center; justify-content: center; gap: 12px; margin-bottom: 20px;">
-                    <img src="./examples/logo.png" style="height: 40px;" />
+                    <img src="{logo_data_url}" style="height: 80px;" />
                     <h1 style="margin: 0; font-size: 32px;">DreamFuse</h1>
                 </div>
                 """