update gradio

app.py

@@ -1,6 +1,5 @@
 from types import MethodType
 
-import spaces
 import os
 import gradio as gr
 import torch
@@ -14,7 +13,7 @@ from huggingface_hub import snapshot_download
 from PIL import Image
 from ip_adapter import StyleShot, StyleContentStableDiffusionControlNetPipeline
 
-device = "cuda" 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 contour_detector = SOFT_HEDdetector()
 lineart_detector = LineartDetector()
@@ -58,7 +57,6 @@ lineart_pipe = StyleContentStableDiffusionControlNetPipeline.from_pretrained(bas
 lineart_styleshot = StyleShot(device, lineart_pipe, lineart_ip_ckpt, lineart_style_aware_encoder_path, lineart_transformer_block_path)
 
 
-@spaces.GPU
 def process(style_image, content_image, prompt, num_samples, image_resolution, condition_scale, style_scale,ddim_steps, guidance_scale, seed, a_prompt, n_prompt, btn1, Contour_Threshold=200):
     weight_dtype = torch.float32
 
@@ -143,7 +141,7 @@ with block:
         with gr.Column():
             style_image = gr.Image(sources=['upload'], type="numpy", label='Style Image')
         with gr.Column():
-            with gr.Box():
+            with gr.Blocks():
                 with gr.Column():
                     content_image = gr.Image(sources=['upload'], type="numpy", label='Content Image (optional)')
                     btn1 = gr.Radio(
@@ -177,16 +175,18 @@ with block:
                                       value='longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality')
         
     with gr.Row():
-        with gr.Box():
-            gr.Markdown("### Results for Contour")
+        gr.Markdown("### Results for Contour")
+    with gr.Row():
+        with gr.Blocks():
             with gr.Row():
                 with gr.Column(scale = 1):
                     contour_gallery = gr.Gallery(label='Contour Output', show_label=True, elem_id="gallery", columns=[1], rows=[1], height='auto')
                 with gr.Column(scale = 4):
                     image_gallery = gr.Gallery(label='Result for Contour', show_label=True, elem_id="gallery", columns=[4], rows=[1], height='auto')
     with gr.Row():
-        with gr.Box():
-            gr.Markdown("### Results for Lineart")
+        gr.Markdown("### Results for Lineart")
+    with gr.Row():
+        with gr.Blocks():
             with gr.Row():
                 with gr.Column(scale = 1):
                     line_gallery = gr.Gallery(label='Lineart Output', show_label=True, elem_id="gallery", columns=[1], rows=[1], height='auto')

requirements.txt

@@ -8,5 +8,5 @@ accelerate==0.21.0
 opencv-python==4.8.1.78
 einops==0.7.0
 botocore==1.34.68
-gradio==3.41.2
+gradio==4.37.2
 basicsr