add 1

URL.TXT

@@ -0,0 +1,4 @@
+https://huggingface.co/spaces/sonygod/GOT-OCR
+https://huggingface.co/spaces/Tonic/GOT-OCR
+
+https://huggingface.co/spaces/sonygod/OnnxTR-OCR

app.py

@@ -12,35 +12,42 @@ def main():
     
     if st.sidebar.button("分析页面"):
         if uploaded_file is None:
-            st.sidebar.write("请上传一个文档")
+            st.sidebar.error("请上传一个文档")
             return
             
         if current_page is None:
-            st.sidebar.write("页面加载失败")
+            st.sidebar.error("页面加载失败")
             return
 
         try:
-            # Process page
-            start_time = time.time()
-            model.load_model(**params)
-            seg_map, out = model.process_page(current_page)
-            
-            # Display results
-            fig, ax = plt.subplots()
-            ax.imshow(seg_map)
-            ax.axis("off")
-            ui.cols[1].pyplot(fig)
-            
-            fig = visualize_page(out.pages[0].export(), out.pages[0].page, 
-                               interactive=False, add_labels=True)
-            ui.cols[2].pyplot(fig)
-            
-            # Display processing time and results
-            total_time = time.time() - start_time
-            st.json({"total_time": total_time, "results": out.pages[0].export()})
+            with st.spinner('正在处理中...请稍等'):
+                st.sidebar.info("开始处理...")
+                # Process page
+                start_time = time.time()
+                st.sidebar.info("加载模型...")
+                model.load_model(**params)
+                st.sidebar.info("分析页面...")
+                seg_map, out = model.process_page(current_page)
+                
+                # Display results
+                st.sidebar.info("显示结果...")
+                fig, ax = plt.subplots()
+                ax.imshow(seg_map)
+                ax.axis("off")
+                ui.cols[1].pyplot(fig)
+                
+                fig = visualize_page(out.pages[0].export(), out.pages[0].page, 
+                                   interactive=False, add_labels=True)
+                ui.cols[2].pyplot(fig)
+                
+                # Display processing time and results
+                total_time = time.time() - start_time
+                st.success(f"处理完成! 用时: {total_time:.2f}秒")
+                st.json({"total_time": total_time, "results": out.pages[0].export()})
             
         except Exception as e:
             st.error(f"处理失败: {str(e)}")
+            st.sidebar.error("发生错误，请查看详细信息")
 
 if __name__ == "__main__":
     main()