Update app.py

app.py

@@ -8,9 +8,18 @@ import os
 import requests
 from tempfile import NamedTemporaryFile
 
-# Load MoveNet model
+
+import tensorflow_hub as hub
+
+
 movenet_model_path = '/models/movenet/movenet_lightning'
-movenet_model = tf.saved_model.load(movenet_model_path)
+
+# Check if the model path exists
+if not os.path.exists(movenet_model_path):
+    # Download the model from TensorFlow Hub
+    movenet_model = hub.load("https://tfhub.dev/google/movenet/singlepose/lightning/4")
+else:
+    movenet_model = tf.saved_model.load(movenet_model_path)
 
 # Load BLIP model
 blip_model = BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-large')