Update app.py

app.py

@@ -37,7 +37,7 @@ class model:
 
             # Instantiate model from config and move to device
             self.model = instantiate_from_config(config.model).to(device)
-            self.model = torch.compile(self.model,mode='reduce-overhead')
+            self.model = torch.compile(self.model,mode='max-autotune')
             
             os.chdir(base_path)