Update app.py

app.py

@@ -1,6 +1,25 @@
-from diffusers import DiffusionPipeline
+import gradio as gr
+import threading
+import numpy as np  # Import NumPy for memory-efficient data structures
+import gc  # Import garbage collector for explicit memory management
 
-pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams_v0.0.5")
+# Increase thread stack size
+threading.stack_size(2**33)
 
-prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
-image = pipe(prompt).images[0]
+# Load the model
+model = gr.load("models/mann-e/Mann-E_Dreams_v0.0.5")
+
+# Preload large datasets or pre-trained weights (if applicable)
+# ...
+
+# Launch the model in a thread
+thread = threading.Thread(target=model.launch)
+
+# Start the thread
+thread.start()
+
+# Explicitly trigger garbage collection to free up memory
+gc.collect()
+
+# Continue with other tasks or image generation code
+# ...