fffiloni commited on
Commit
d35d17d
1 Parent(s): 212387e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -18,7 +18,7 @@ def infer(video_frames, masks_frames):
18
  os.makedirs(output_folder)
19
 
20
 
21
- bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
22
 
23
  command = [
24
  f"python",
@@ -33,7 +33,13 @@ def infer(video_frames, masks_frames):
33
  # Get the list of files in the "results" folder
34
  result_files = os.listdir(bmx_trees_folder)
35
 
36
- return result_files
 
 
 
 
 
 
37
 
38
  css="""
39
  #col-container{
@@ -60,7 +66,7 @@ with gr.Blocks(css=css) as demo:
60
  submit_btn = gr.Button("Submit")
61
 
62
  with gr.Column():
63
- result = gr.Files(label="Result")
64
 
65
 
66
 
 
18
  os.makedirs(output_folder)
19
 
20
 
21
+ #bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
22
 
23
  command = [
24
  f"python",
 
33
  # Get the list of files in the "results" folder
34
  result_files = os.listdir(bmx_trees_folder)
35
 
36
+ # Print the content of the "results/bmx-trees" folder
37
+ print("Contents of the results folder:")
38
+ for item in output_folder:
39
+ print(item)
40
+
41
+
42
+ return "done"
43
 
44
  css="""
45
  #col-container{
 
66
  submit_btn = gr.Button("Submit")
67
 
68
  with gr.Column():
69
+ result = gr.Textbox(label="Result")
70
 
71
 
72