Update app.py

app.py

@@ -1,5 +1,16 @@
 import gradio as gr
 
+names = []
+for name in os.listdir(weight_root):
+    if name.endswith(".pth"):
+        names.append(name)
+index_paths = []
+for root, dirs, files in os.walk(index_root, topdown=False):
+    for name in files:
+        if name.endswith(".index") and "trained" not in name:
+            index_paths.append("%s/%s" % (root, name))
+
+
 def change_choices():
     names = []
     for name in os.listdir(weight_root):