mrneuralnet commited on
Commit
7576905
1 Parent(s): 63c711f

Fix video inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def main():
67
 
68
 
69
  with st.spinner(f'Analyzing {file_type}...'):
70
- if file_type == 'video':
71
  result = video_pipeline(video_path, config_payload=config_payload)
72
  else:
73
  result = image_pipeline({'images': [img2base64(np.array(img))]}, config_payload=config_payload)
 
67
 
68
 
69
  with st.spinner(f'Analyzing {file_type}...'):
70
+ if file_type == 'video' or file_type == 'mp4':
71
  result = video_pipeline(video_path, config_payload=config_payload)
72
  else:
73
  result = image_pipeline({'images': [img2base64(np.array(img))]}, config_payload=config_payload)