Update app.py

app.py

@@ -6,10 +6,13 @@ import gc
 import psutil
 import time
 
-# Enable garbage collection and set deterministic behavior
+# Enable garbage collection
 gc.enable()
-torch.backends.cudnn.benchmark = False if torch.cuda.is_available() else None
-torch.backends.cudnn.deterministic = True if torch.cuda.is_available() else None
+
+# Properly handle CUDA settings
+if torch.cuda.is_available():
+    torch.backends.cudnn.benchmark = False
+    torch.backends.cudnn.deterministic = True
 
 # Global variables
 model = None