Update gradio_seesr_turbo.py

gradio_seesr_turbo.py

@@ -47,6 +47,11 @@ snapshot_download(
 )
 
 
+snapshot_download(
+    repo_id="xinyu1205/recognize_anything_model", 
+    local_dir="preset/models/"
+)
+
 # Load scheduler, tokenizer and models.
 pretrained_model_path = 'preset/models/sd-turbo'
 seesr_model_path = 'preset/models/seesr'
@@ -56,7 +61,7 @@ text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="t
 tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
 vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")
 # feature_extractor = CLIPImageProcessor.from_pretrained(f"{pretrained_model_path}/feature_extractor")
-unet = UNet2DConditionModel.from_pretrained_orig(seesr_model_path, subfolder="unet")
+unet = UNet2DConditionModel.from_pretrained_orig(pretrained_model_path, seesr_model_path, subfolder="unet")
 controlnet = ControlNetModel.from_pretrained(seesr_model_path, subfolder="controlnet")
 
 # Freeze vae and text_encoder
@@ -181,8 +186,8 @@ with block:
         gr.Markdown(MARKDOWN)
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(source="upload", type="pil")
-            run_button = gr.Button(label="Run")
+            input_image = gr.Image(type="pil")
+            run_button = gr.Button("Run")
             with gr.Accordion("Options", open=True):
                 user_prompt = gr.Textbox(label="User Prompt", value="")
                 positive_prompt = gr.Textbox(label="Positive Prompt", value="clean, high-resolution, 8k, best quality, masterpiece")
@@ -198,7 +203,7 @@ with block:
                 latent_tiled_overlap = gr.Slider(label="Diffusion Tile Overlap", minimum=4, maximum=16, value=4, step=1)
                 scale_factor = gr.Number(label="SR Scale", value=4)
         with gr.Column():
-            result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery").style(grid=2, height="auto")
+            result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery")
 
     inputs = [
         input_image,
@@ -215,5 +220,5 @@ with block:
     ]
     run_button.click(fn=process, inputs=inputs, outputs=[result_gallery])
 
-block.launch()
+block.launch(share=True)