pablorodriper commited on
Commit
5136061
·
1 Parent(s): cb525f9

Refactor repository

Browse files
app.py CHANGED
@@ -2,12 +2,11 @@ import glob
2
 
3
  import gradio as gr
4
  import tensorflow as tf
5
- from huggingface_hub import from_pretrained_keras
6
 
7
- from predict import predict_label
8
 
9
  ##Create list of examples to be loaded
10
- example_list = glob.glob("*.mp4")
11
  example_list = list(map(lambda el:[el], example_list))
12
 
13
  demo = gr.Blocks()
@@ -35,7 +34,7 @@ with demo:
35
 
36
  submit_button.click(predict_label, inputs=input_video, outputs=output_label)
37
 
38
- gr.Markdown('\n Demo created by: <a href=\"https://huggingface.co/pablorodriper\"> Pablo Rodríguez</a> Based on the Keras example by <a href=\"https://keras.io/examples/vision/vivit/\">Aritra Roy Gosthipaty and Ayush Thakur</a>')
39
 
40
  demo.launch()
41
 
 
2
 
3
  import gradio as gr
4
  import tensorflow as tf
 
5
 
6
+ from utils.predict import predict_label
7
 
8
  ##Create list of examples to be loaded
9
+ example_list = glob.glob("examples/*.mp4")
10
  example_list = list(map(lambda el:[el], example_list))
11
 
12
  demo = gr.Blocks()
 
34
 
35
  submit_button.click(predict_label, inputs=input_video, outputs=output_label)
36
 
37
+ gr.Markdown('\n Demo created by: <a href=\"https://huggingface.co/pablorodriper\"> Pablo Rodríguez</a>; Based on the Keras example by <a href=\"https://keras.io/examples/vision/vivit/\">Aritra Roy Gosthipaty and Ayush Thakur</a>')
38
 
39
  demo.launch()
40
 
example_1.mp4 → examples/example_1.mp4 RENAMED
File without changes
utils/__init__.py ADDED
File without changes
constants.py → utils/constants.py RENAMED
File without changes
predict.py → utils/predict.py RENAMED
File without changes