modifying requirements.txt

gradio_app.py

@@ -4,7 +4,7 @@ from datetime import datetime
 from pathlib import Path
 import gradio as gr
 import numpy as np
-
+from huggingface_hub import hf_hub_download, snapshot_download
 # -----------------------------
 # Setup paths and env
 # -----------------------------
@@ -13,6 +13,9 @@ os.environ["HF_HOME"] = HF_HOME
 os.environ["TRANSFORMERS_CACHE"] = HF_HOME
 os.makedirs(HF_HOME, exist_ok=True)
 
+hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")
+snapshot_download(repo_id="AlexWortega/RIFE", local_dir="model_rife")
+
 PRETRAINED_DIR = "/app/pretrained"
 os.makedirs(PRETRAINED_DIR, exist_ok=True)