Upload 12 files

__init__.py

@@ -0,0 +1,4 @@
+from .app import MinerUModel
+from .pipeline import MinerUPipeline
+
+__version__ = "1.0.0" 

app.py

@@ -6,15 +6,22 @@ class MinerUModel(PreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
         self.config = config
+        self._setup_models()
         
-    @staticmethod
-    def load_model():
+    def _setup_models(self):
         from model_loader import MinerUModelLoader
-        return MinerUModelLoader.load_models("./")
+        self.models = MinerUModelLoader.load_models("./")
+        
+    @classmethod
+    def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
+        config = kwargs.pop("config", None)
+        model = cls(config)
+        model._setup_models()
+        return model
         
     def forward(self, input_data):
         # 实现前向传播逻辑
-        pass
+        return self.models["layout"](input_data)
 
 def load_model():
     model = MinerUModel.from_pretrained("./")

model-card.yml

@@ -0,0 +1,11 @@
+---
+language:
+  - zh
+  - en
+license: apache-2.0
+library_name: transformers
+pipeline_tag: document-conversion
+tags:
+  - pdf-to-markdown
+  - document-conversion
+--- 

model_index.json

@@ -3,5 +3,8 @@
   "architectures": ["MinerUModel"],
   "model_type": "mineru",
   "transformers_version": "4.28.0",
-  "framework": "pytorch"
+  "framework": "pytorch",
+  "task": "document-conversion",
+  "pipeline_class": "MinerUPipeline",
+  "auto_class": "AutoModelForDocumentConversion"
 } 

special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "pad_token": "[PAD]",
+  "unk_token": "[UNK]",
+  "cls_token": "[CLS]",
+  "sep_token": "[SEP]"
+}