add requirements

app.py

@@ -14,7 +14,7 @@ import time
 from queue import Queue
 
 
-os.environ["CUDA_VISIBLE_DEVICES"] = '0'
+# os.environ["CUDA_VISIBLE_DEVICES"] = '0'
 
 # 初始化模型
 print("Loading model...")
@@ -26,7 +26,7 @@ model = AutoModel.from_pretrained(
     trust_remote_code=True,
     use_safetensors=True
 )
-model = model.eval().cuda().to(torch.bfloat16)
+model = model.eval()  # .cuda().to(torch.bfloat16)
 print("Model loaded successfully!")
 
 

requirements.txt

@@ -0,0 +1,9 @@
+transformers==4.46.3
+tokenizers==0.20.3
+PyMuPDF
+img2pdf
+einops
+easydict
+addict 
+Pillow
+numpy