Spaces:
Sleeping
Sleeping
feat: add requirements.txt and fetch MoveNet Thunder model from TensorFlow Hub
Browse files- download_movenet.py +9 -0
- 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
|