fix pose im size

app.py

@@ -13,7 +13,7 @@ def update_model_choices(task):
     model_choices = list(SAPIENS_LITE_MODELS_PATH[task.lower()].keys())
     return gr.Dropdown(choices=model_choices, value=model_choices[0] if model_choices else None)
 
-@spaces.GPU()
+@spaces.GPU(duration=12)
 def process_image(input_image, task, version):
     if isinstance(input_image, np.ndarray):
         input_image = Image.fromarray(input_image)

inference/pose.py

@@ -137,6 +137,8 @@ def process_image_or_video(input_data, task='pose', version='sapiens_1b'):
     if model is None or device is None:
         return None
 
+    input_shape = (3, 1024, 768)
+
     def process_frame(frame):
         if isinstance(frame, np.ndarray):
             frame = Image.fromarray(frame)
@@ -144,8 +146,6 @@ def process_image_or_video(input_data, task='pose', version='sapiens_1b'):
         if frame.mode == 'RGBA':
             frame = frame.convert('RGB')
         
-        input_shape = (3, frame.height, frame.width)
-        
         img = preprocess_image(frame, input_shape)
         
         with torch.no_grad():