Nick088 commited on
Commit
61c9fcc
1 Parent(s): bdeb650

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -29
app.py CHANGED
@@ -72,41 +72,18 @@ def inference_video(video, size):
72
  video_mp4_result_folder = 'results_mp4_videos'
73
  result_restored_imgs_folder = 'restored_imgs'
74
 
75
- if os.path.isdir(upload_folder):
76
- print(upload_folder+" exists")
77
- else:
78
- os.makedirs(upload_folder, exist_ok=True)
79
 
80
- if os.path.isdir(video_folder):
81
- print(video_folder+" exists")
82
- else:
83
- os.makedirs(video_folder, exist_ok=True)
84
 
85
- if os.path.isdir(video_result_folder):
86
- print(video_result_folder+" exists")
87
- else:
88
- os.makedirs(video_result_folder, exist_ok=True)
89
 
90
- if os.path.isdir(video_mp4_result_folder):
91
- print(video_mp4_result_folder+" exists")
92
- else:
93
- os.makedirs(video_mp4_result_folder, exist_ok=True)
94
 
95
- if os.path.isdir(result_folder):
96
- print(result_folder+" exists")
97
- else:
98
- os.makedirs(result_folder, exist_ok=True)
99
 
100
  os.chdir("results")
101
- if os.path.isdir(result_restored_imgs_folder):
102
- print(result_restored_imgs_folder+" exists")
103
- else:
104
- os.makedirs(result_restored_imgs_folder, exist_ok=True)
105
- os.chdir("..")
106
-
107
- if os.path.isdir(video_folder):
108
- shutil.rmtree(video_folder)
109
- os.makedirs(video_folder, exist_ok=True)
110
  os.chdir("..")
111
  try:
112
  # Specify the desired output file path with the custom name and ".mp4" extension
 
72
  video_mp4_result_folder = 'results_mp4_videos'
73
  result_restored_imgs_folder = 'restored_imgs'
74
 
75
+ os.makedirs(upload_folder, exist_ok=True)
 
 
 
76
 
77
+ os.makedirs(video_folder, exist_ok=True)
 
 
 
78
 
79
+ os.makedirs(video_result_folder, exist_ok=True)
 
 
 
80
 
81
+ os.makedirs(video_mp4_result_folder, exist_ok=True)
 
 
 
82
 
83
+ os.makedirs(result_folder, exist_ok=True)
 
 
 
84
 
85
  os.chdir("results")
86
+ os.makedirs(result_restored_imgs_folder, exist_ok=True)
 
 
 
 
 
 
 
 
87
  os.chdir("..")
88
  try:
89
  # Specify the desired output file path with the custom name and ".mp4" extension