Spaces:
Sleeping
Sleeping
Update src/predict.py
Browse files- src/predict.py +2 -0
src/predict.py
CHANGED
|
@@ -2,6 +2,8 @@ from tensorflow.keras.models import load_model
|
|
| 2 |
|
| 3 |
#def load_trained_model(path):
|
| 4 |
#return load_model(path)
|
|
|
|
|
|
|
| 5 |
|
| 6 |
|
| 7 |
def run_prediction(model, processed_frame): # Renamed function
|
|
|
|
| 2 |
|
| 3 |
#def load_trained_model(path):
|
| 4 |
#return load_model(path)
|
| 5 |
+
# ✅ Set writable cache directory
|
| 6 |
+
os.environ["HF_HUB_CACHE"] = "/tmp/huggingface"
|
| 7 |
|
| 8 |
|
| 9 |
def run_prediction(model, processed_frame): # Renamed function
|