babypoby commited on
Commit
0587d81
·
1 Parent(s): 9372a94
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -74,6 +74,7 @@ def greet(image_path: str):
74
 
75
  for file_name in os.listdir(output_folder):
76
  file_path = os.path.join(output_folder, file_name)
 
77
  probs = classify(file_path)
78
  top_3 = probs.head(3)
79
  top_3_list = [[cls, prob] for cls, prob in top_3.items()]
 
74
 
75
  for file_name in os.listdir(output_folder):
76
  file_path = os.path.join(output_folder, file_name)
77
+ print(file_path)
78
  probs = classify(file_path)
79
  top_3 = probs.head(3)
80
  top_3_list = [[cls, prob] for cls, prob in top_3.items()]