update requirements

app.py

@@ -7,7 +7,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_name = "Lin-Chen/ShareCaptioner"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
-    model_name, device_map="cuda", trust_remote_code=True).eval()
+    model_name, device_map="cpu", trust_remote_code=True).eval()
 model.tokenizer = tokenizer
 
 model.cuda()

requirements.txt

@@ -1,14 +1,29 @@
-transformers==4.32.0
-accelerate==0.24.0
-tiktoken==0.5.1
-einops==0.7.0
-transformers_stream_generator==0.0.4
-scipy==1.11.3
-torchvision==0.15.2
-pillow==10.0.1
-matplotlib==3.8.0
-gradio==3.50.2
-sentencepiece
-urllib3==1.26.18
-timm==0.6.13
+torch==2.1.2
+torchvision==0.16.2
+transformers==4.37.2
+tokenizers==0.15.1
+sentencepiece==0.1.99
+accelerate==0.21.0
+peft
+bitsandbytes
+pydantic
+markdown2[all]
+numpy
+scikit-learn==1.2.2
+gradio==4.16.0
+gradio_client==0.8.1
+openai
 spaces
+requests
+httpx==0.24.0
+uvicorn
+fastapi
+decord
+einops==0.6.1
+einops-exts==0.0.4
+timm==0.6.13
+deepspeed==0.12.6
+ninja
+wandb
+build
+twine