modified: README.md

README.md

@@ -5,6 +5,7 @@ colorFrom: gray
 colorTo: purple
 sdk: gradio
 sdk_version: 5.0.1
+python_version: 3.12.9
 app_file: app.py
 pinned: false
 license: apache-2.0

app.py

@@ -86,6 +86,10 @@ class FluxEditor:
         self.output_dir = os.path.join(out_root, name_dir)
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)
+        if not os.path.exists("heatmap"):
+            os.makedirs("heatmap")
+        if not os.path.exists("heatmap/average_heatmaps"):
+            os.makedirs("heatmap/average_heatmaps")
         self.instructions = ['source']
         self.source_image = None
         self.history_tensors = {

src/fireflow

@@ -0,0 +1 @@
+Subproject commit df4eab3f73eed5efa175438532962c42bf033cf2