Update app.py

app.py

@@ -15,6 +15,13 @@ import torch.nn.functional as F
 device = torch.device('cpu')
 torch.set_num_threads(4)  # Optimize CPU performance
 
+# Optimize memory usage
+torch.backends.cudnn.benchmark = False
+torch.backends.cudnn.deterministic = True
+
+# Reduce memory usage for history
+MAX_HISTORY_ENTRIES = 5
+
 # Style presets
 STYLE_PRESETS = {
     "Sketch": {"line_thickness": 1.0, "contrast": 1.2, "brightness": 1.0},
@@ -62,8 +69,8 @@ class HistoryManager:
             return self.history[-1]["settings"]
         return None
 
-# Initialize history manager
-history_manager = HistoryManager()
+# Initialize history manager with reduced entries
+history_manager = HistoryManager(max_entries=MAX_HISTORY_ENTRIES)
 
 norm_layer = nn.InstanceNorm2d
 
@@ -401,10 +408,14 @@ with gr.Blocks(css=custom_css) as iface:
         ]
     )
 
-# Launch the interface
+# Launch the interface with optimized settings
 iface.launch(
     server_name="0.0.0.0",
     server_port=7860,
     share=False,
-    debug=False
+    debug=False,
+    show_error=True,
+    max_threads=4,
+    ssr=False,  # Disable SSR to prevent Node.js issues
+    cache_examples=False,  # Disable example caching to save memory
 )