Update app.py

app.py

@@ -1,17 +1,25 @@
 import gradio as gr
-import concurrent.futures
+import threading
+import numpy as np  # Import NumPy for memory-efficient data structures
+import gc  # Import garbage collector for explicit memory management
 
-# Load the model into RAM
+# Increase thread stack size
+threading.stack_size(2**33)
+
+# Load the model
 model = gr.load("models/mann-e/Mann-E_Dreams")
 
-def interact(input):
-    # Define the function for user interaction
-    response = model(input)
-    return response
+# 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()
 
-# Use ThreadPoolExecutor to manage the threads
-with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
-    # Create a Gradio interface with the loaded model
-    interface = gr.Interface(fn=interact, inputs="text", outputs="image")
-    # Handle the interactions with Gradio
-    interface.launch() 
+# Continue with other tasks or image generation code
+# ...