aiface commited on
Commit
7c5fb79
1 Parent(s): 7b5cf19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -142,10 +142,13 @@ def preprocess_video(input_video_path):
142
 
143
  def predict(process_video):
144
  os.chdir('/home/user/app')
145
- os.system('bash TestVisual.sh')
146
- pred=[]
147
- fi = open("/home/user/app/result/ho.txt", 'r')
148
- pred = fi.readlines()
 
 
 
149
  return pred[-1]
150
 
151
 
 
142
 
143
  def predict(process_video):
144
  os.chdir('/home/user/app')
145
+ try:
146
+ os.system('bash ./TestVisual.sh')
147
+ pred=[]
148
+ fi = open("/home/user/app/result/ho.txt", 'r')
149
+ pred = fi.readlines()
150
+ except IndexError:
151
+ pred=['error']
152
  return pred[-1]
153
 
154