Update app.py

app.py

@@ -174,50 +174,21 @@ tag_model.eval()
 tag_model.to(device, dtype=weight_dtype)
 
 def preprocess_image(input_image: Image.Image) -> Image.Image:
-
-    ori_width, ori_height = input_image.size
-
-    duration = 60
-
-    if ori_width > 384 or ori_height > 384:
-        duration = 180
-    elif ori_width > 1024 or ori_height > 1024:
-        duration = 240
-
-    return duration
+    img = input_image.copy()
+    img.thumbnail((256, 256), Image.Resampling.BILINEAR)
+    return img
 
 @spaces.GPU()
 def preprocess_n_magnify(input_image: Image.Image):
-    duration = preprocess_image(input_image)
-
-    img, magnified_img = magnify(input_image, duration)    
+    processed_img = preprocess_image(input_image)
 
-    return duration, (img, magnified_img)
+    img, magnified_img = magnify(processed_img)    
 
+    return (img, magnified_img)
 
-def get_duration(
-    input_image: Image.Image,
-    duration_seconds,
-    user_prompt,
-    positive_prompt,
-    negative_prompt,
-    num_inference_steps,
-    scale_factor,
-    cfg_scale,
-    seed,
-    latent_tiled_size,
-    latent_tiled_overlap,
-    sample_times,
-    progress,
-    ):
-    
-    return int(duration_seconds)
-        
-    
-@spaces.GPU(duration=get_duration)
+@spaces.GPU()
 def magnify(
     input_image: Image.Image,
-    duration_seconds = 60,
     user_prompt = "",
     positive_prompt = "clean, high-resolution, 8k, best quality, masterpiece",
     negative_prompt = "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
@@ -360,14 +331,14 @@ with gr.Blocks(css=css, theme=theme) as demo:
                     inputs=[
                         input_image,
                     ],
-                    outputs=[duration_time, result_gallery],
+                    outputs=[result_gallery],
                     fn=preprocess_n_magnify,
                     cache_examples=True,
                 )
         inputs = [
             input_image,
         ]
-        run_button.click(fn=magnify, inputs=[input_image,duration_time], outputs=[result_gallery])
-        input_image.upload(fn=preprocess_image,inputs=input_image, outputs=duration_time)
+        run_button.click(fn=magnify, inputs=input_image, outputs=[result_gallery])
+        input_image.upload(fn=preprocess_image,inputs=input_image, outputs=input_image)
 
 demo.launch(share=True)