fixing code

app.py

@@ -1,5 +1,6 @@
-import os 
+import os
 import numpy as np
+import base64
 import io
 import requests
 from io import BytesIO
@@ -43,12 +44,19 @@ def get_dark_mode():
         document.body.classList.toggle('dark');
     }
     """
+    
+def pil_image_to_data_uri(img, format="PNG"):
+        buffered = BytesIO()
+        img.save(buffered, format=format)
+        img_str = base64.b64encode(buffered.getvalue()).decode()
+        return f"data:image/{format.lower()};base64,{img_str}"
 
 def clear_session(image):
     global clear_flag 
     clear_flag = True
     ones = Image.new("L", (1024, 1024), 255)
-    return ones, gr.update(value=None)
+    temp_uri = pil_image_to_data_uri(ones)
+    return gr.update(link=temp_uri), gr.update(value=None)
 
 
 def assign_gpu(options, img_init, text_init, prompt_template_init, style_name_init, seeds_init, val_r_values_init, faster_init, model_name_init, input_type):