Update app.py

app.py

@@ -7,8 +7,8 @@ from huggingface_hub import HfApi
 from io import StringIO
 from transformers import pipeline
 import torch
-import requests # Needed for downloading models
-from tqdm import tqdm # For download progress bar
+import requests
+from tqdm import tqdm
 import spaces
 import functools
 from constants import updated_upscaler_dict as UPSCALER_DICT_GUI
@@ -19,7 +19,6 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DIRECTORY_UPSCALERS = "upscalers"
 
 # --- Configuration ---
-# Set your Hugging Face Write Token as an environment variable
 # export HF_TOKEN_ORG="hf_YourTokenHere"
 HF_TOKEN_ORG = os.getenv("HF_TOKEN_ORG")
 DATASET_REPO_ID = "TestOrganizationPleaseIgnore/upscale_board_data"
@@ -337,7 +336,7 @@ class UpscalerApp:
                 with gr.Row():
                     with gr.Column(scale=1):
                         input_image_playground = gr.Image(type="pil", label="Source Image", format="png")
-                        upscaler_model_dropdown = gr.Dropdown(value="8x_NMKD-Superscale_150000_G", choices=list(UPSCALER_DICT_GUI.keys()), label="Upscaler Model")
+                        upscaler_model_dropdown = gr.Dropdown(value="R-ESRGAN_4x+", choices=list(UPSCALER_DICT_GUI.keys()), label="Upscaler Model")
                         run_button_playground = gr.Button("Run Upscale")
                     with gr.Column(scale=2):
                         output_image_playground = gr.Image(label="Upscaled Result", interactive=False, format="png")
@@ -361,9 +360,7 @@ def gpu_tab1(*args, **kwargs):
 def gpu_tab2(*args, **kwargs):
     return app.playground_upscale(*args, **kwargs)
 
-# --- Main Execution Block ---
 if __name__ == "__main__":
-    # Before launching, ensure the upscalers directory exists
     if not os.path.exists(DIRECTORY_UPSCALERS):
         os.makedirs(DIRECTORY_UPSCALERS)