Rishabh12j commited on
Commit
90b7da2
·
verified ·
1 Parent(s): 51d0667

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ css = """
42
 
43
 
44
  def checkpoint_list():
45
- return [model.name for model in BASE_FOLDER.glob("*")]
46
 
47
 
48
 
 
42
 
43
 
44
  def checkpoint_list():
45
+ return [model.name for model in BASE_FOLDER.glob("*.ckpt")]
46
 
47
 
48