lukassso commited on
Commit
3607858
·
1 Parent(s): 9feaf34

feat: add requirements.txt and fetch MoveNet Thunder model from TensorFlow Hub

Browse files
Files changed (2) hide show
  1. download_movenet.py +9 -0
  2. requirements.txt +5 -0
download_movenet.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import tensorflow as tf
2
+ import tensorflow_hub as hub
3
+ import os
4
+
5
+ save_path = "movenet_saved_model"
6
+ model_url = "https://tfhub.dev/google/movenet/singlepose/thunder/4"
7
+ model = hub.load(model_url)
8
+ tf.saved_model.save(model, export_dir=save_path)
9
+ print(f"Model saved to: {os.path.abspath(save_path)}")
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ tensorflow==2.15.0
2
+ gradio
3
+ numpy
4
+ Pillow
5
+ huggingface_hub