feat: update new 4xRRDB weight option

app.py

@@ -21,6 +21,10 @@ def auto_download_if_needed(weight_path):
     if not os.path.exists("pretrained"):
         os.makedirs("pretrained")
     
+    if weight_path == "pretrained/4x_APISR_RRDB_GAN_generator.pth":
+        os.system("wget https://github.com/Kiteretsu77/APISR/releases/download/v0.2.0/4x_APISR_RRDB_GAN_generator.pth")
+        os.system("mv 4x_APISR_RRDB_GAN_generator.pth pretrained")
+    
     if weight_path == "pretrained/4x_APISR_GRL_GAN_generator.pth":
         os.system("wget https://github.com/Kiteretsu77/APISR/releases/download/v0.1.0/4x_APISR_GRL_GAN_generator.pth")
         os.system("mv 4x_APISR_GRL_GAN_generator.pth pretrained")
@@ -29,6 +33,7 @@ def auto_download_if_needed(weight_path):
         os.system("wget https://github.com/Kiteretsu77/APISR/releases/download/v0.1.0/2x_APISR_RRDB_GAN_generator.pth")
         os.system("mv 2x_APISR_RRDB_GAN_generator.pth pretrained")
 
+    
 
 
 def inference(img_path, model_name):
@@ -42,6 +47,11 @@ def inference(img_path, model_name):
             auto_download_if_needed(weight_path)
             generator = load_grl(weight_path, scale=4)  # Directly use default way now
             
+        elif model_name == "4xRRDB":
+            weight_path = "pretrained/4x_APISR_RRDB_GAN_generator.pth"
+            auto_download_if_needed(weight_path)
+            generator = load_rrdb(weight_path, scale=4)  # Directly use default way now
+            
         elif model_name == "2xRRDB":
             weight_path = "pretrained/2x_APISR_RRDB_GAN_generator.pth"
             auto_download_if_needed(weight_path)
@@ -80,7 +90,8 @@ if __name__ == '__main__':
     APISR aims at restoring and enhancing low-quality low-resolution anime images and video sources with various degradations from real-world scenarios.
 
     ### Note: Due to memory restriction, all images whose short side is over 720 pixel will be downsampled to 720 pixel with the same aspect ratio.  E.g., 1920x1080 -> 1280x720
-            
+    ### Note: Please check [Model Zoo](https://github.com/Kiteretsu77/APISR/blob/main/docs/model_zoo.md) for the description of each weight.
+    
     If APISR is helpful, please help star the GitHub Repo. Thanks! 
     """
 
@@ -94,6 +105,7 @@ if __name__ == '__main__':
                 model_name = gr.Dropdown(
                     [
                         "2xRRDB",
+                        "4xRRDB",
                         "4xGRL"
                     ],
                     type="value",