remove safetensors

app.py

@@ -17,7 +17,7 @@ else:
     from transformers import AutoModel, AutoTokenizer
     tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
     try:
-        model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, use_safetensors=True, pad_token_id=tokenizer.eos_token_id)
+        model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, pad_token_id=tokenizer.eos_token_id)
         model = model.eval()
     except Exception as e:
         # If model fails to load (e.g. due to no network or heavy resources), keep a placeholder

requirements.txt

@@ -3,6 +3,5 @@ transformers
 torch
 Pillow
 numpy
-safetensorsgradio
 pillow
 numpy