aifeifei798 commited on
Commit
b605bce
1 Parent(s): 6233830

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,12 +7,12 @@ from PIL import Image
7
  import subprocess
8
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
9
 
10
- model = AutoModelForCausalLM.from_pretrained('John6666/gokaygokay-Florence-2-SD3-Captioner-8bit', trust_remote_code=True).to("cpu").eval()
11
 
12
- processor = AutoProcessor.from_pretrained('John6666/gokaygokay-Florence-2-SD3-Captioner-8bit', trust_remote_code=True)
13
 
14
 
15
- TITLE = "# [Florence-2 SD3 Long Captioner 8bit](https://huggingface.co/John6666/gokaygokay-Florence-2-SD3-Captioner-8bit)"
16
  DESCRIPTION = "[Florence-2 Base](https://huggingface.co/microsoft/Florence-2-base-ft) fine-tuned on Long SD3 Prompt and Image pairs. Check above link for datasets that are used for fine-tuning."
17
 
18
  def modify_caption(caption: str) -> str:
 
7
  import subprocess
8
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
9
 
10
+ model = AutoModelForCausalLM.from_pretrained('gokaygokay/Florence-2-SD3-Captioner', trust_remote_code=True).to("cpu").eval()
11
 
12
+ processor = AutoProcessor.from_pretrained('gokaygokay/Florence-2-SD3-Captioner', trust_remote_code=True)
13
 
14
 
15
+ TITLE = "# [Florence-2 SD3 Long Captioner](https://huggingface.co/gokaygokay/Florence-2-SD3-Captioner/)"
16
  DESCRIPTION = "[Florence-2 Base](https://huggingface.co/microsoft/Florence-2-base-ft) fine-tuned on Long SD3 Prompt and Image pairs. Check above link for datasets that are used for fine-tuning."
17
 
18
  def modify_caption(caption: str) -> str: