Update app.py

app.py

@@ -1,25 +1,17 @@
 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
+import concurrent.futures
 
-# Increase thread stack size
-threading.stack_size(2**33)
-
-# Load the model
+# Load the model into RAM
 model = gr.load("models/mann-e/Mann-E_Dreams")
 
-# 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()
+def interact(input):
+    # Define the function for user interaction
+    response = model(input)
+    return response
 
-# Continue with other tasks or image generation code
-# ...  
+# 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()