thwri commited on
Commit
9f2d6e6
·
verified ·
1 Parent(s): a9e8fe0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
12
  model = AutoModelForCausalLM.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True).eval()#.to("cuda").eval()
13
  processor = AutoProcessor.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True)
14
 
15
- TITLE = "# [thwri/CogFlorence-2.1-Large]"
16
- DESCRIPTION = "microsoft/Florence-2-large tuned on Ejafa/ye-pop captioned with CogVLM2"
17
 
18
  def modify_caption(caption: str) -> str:
19
  special_patterns = [
 
12
  model = AutoModelForCausalLM.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True).eval()#.to("cuda").eval()
13
  processor = AutoProcessor.from_pretrained('thwri/CogFlorence-2.1-Large', trust_remote_code=True)
14
 
15
+ TITLE = "# [thwri/CogFlorence-2.1-Large](https://huggingface.co/thwri/CogFlorence-2.1-Large/)"
16
+ DESCRIPTION = "[microsoft/Florence-2-large](https://huggingface.co/microsoft/Florence-2-large) tuned on [Ejafa/ye-pop](https://huggingface.co/datasets/Ejafa/ye-pop) captioned with [CogVLM2](https://huggingface.co/THUDM/cogvlm2-llama3-chat-19B)"
17
 
18
  def modify_caption(caption: str) -> str:
19
  special_patterns = [