fix bug

basicsr/__init__.py

@@ -8,4 +8,4 @@ from .models import *
 from .ops import *
 from .train import *
 from .utils import *
-from .version import __gitsha__, __version__
+# from .version import __gitsha__, __version__

hugging_face/app.py

@@ -239,9 +239,12 @@ download_url_to_file("https://github.com/jnjaby/KEEP/releases/download/media/rea
 download_url_to_file("https://github.com/jnjaby/KEEP/releases/download/media/real_3.mp4", os.path.join(sample_videos_dir, "real_3.mp4"))
 download_url_to_file("https://github.com/jnjaby/KEEP/releases/download/media/real_4.mp4", os.path.join(sample_videos_dir, "real_4.mp4"))
 
-model_dir = os.path.join("../weights/KEEP")
+model_dir = os.path.join("./weights/KEEP")
 _ = load_file_from_url(url='https://github.com/jnjaby/KEEP/releases/download/v1.0.0/KEEP-b76feb75.pth', model_dir=model_dir, progress=True, file_name=None)
 
+realesrgan_dir = os.path.join("./weights/realesrgan")
+_ = load_file_from_url(url='https://github.com/jnjaby/KEEP/releases/download/v1.0.0/RealESRGAN_x2plus.pth', model_dir=model_dir, progress=True, file_name=None)
+
 # Launching the Gradio interface.
 demo = gr.Interface(
     fn=process_video,