Update app.py

app.py

@@ -666,8 +666,10 @@ def get_latest_checkpoint():
     checkpoints = [
         filename
         for filename in files
-        if "codva2_sft" in filename.lower()
-        and filename.endswith(".pt")
+        if filename.endswith(".pt")
+        and "config" not in filename
+        and "final" not in filename
+        and re.search(r"step[_-]?\d+", filename, re.IGNORECASE)
     ]
 
     if not checkpoints: