Update app.py

app.py

@@ -153,7 +153,7 @@ def analyze_and_visualize_layout(input_image: Image.Image, selected_model_name:
 
         progress(0.5, desc=f"Generating layout data with {model_name_desc}...")
         with torch.no_grad():
-            output_ids = model_to_run.generate(**inputs, max_new_tokens=4096, do_sample=False)
+            output_ids = model_to_run.generate(**inputs, max_new_tokens=4096*2, do_sample=False)
         
         raw_text = processor.batch_decode(output_ids[:, inputs.input_ids.shape[1]:], skip_special_tokens=True)[0]
         
@@ -210,7 +210,18 @@ def analyze_and_visualize_layout(input_image: Image.Image, selected_model_name:
     except IOError:
         font = ImageFont.load_default()
 
-    for item in sorted(results, key=lambda x: x.get("order", 999)):
+        # 推荐的修改
+    def get_safe_order(item):
+        """一个安全的函数，用于获取并转换order值为整数，处理潜在的错误。"""
+        try:
+            # 尝试将获取到的值转换为整数
+            return int(item.get("order", 999))
+        except (ValueError, TypeError):
+            # 如果转换失败（例如，值是"abc"这样的非数字字符串），则返回默认值
+            return 999
+    
+    # 在排序时使用这个新函数
+    for item in sorted(results, key=get_safe_order):
         bbox, label, order = item.get("bbox_2d"), item.get("label", "other"), item.get("order", "")
         if not bbox or len(bbox) != 4: continue