Spaces:

yfyangd
/

TaiwanFrog

Runtime error

yfyangd commited on Aug 14, 2022

Commit

97b1b16

1 Parent(s): 4c34810

Update DataGenerator.py

Files changed (1) hide show

DataGenerator.py CHANGED Viewed

@@ -18,8 +18,8 @@ import matplotlib.pyplot as plt
 def Data_Generator():
     list_test = []
-    for j in os.listdir('./test/'):
-        list_test.append('./test/'+j)
     fname = list_test[0]
     return fname

 def Data_Generator():
     list_test = []
+    for j in os.listdir('./'):
+        list_test.append('./'+j)
     fname = list_test[0]
     return fname