aifeifei798 commited on
Commit
f263b93
1 Parent(s): 5061c3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,9 +9,9 @@ import subprocess
9
  subprocess.run('pip install einops', shell=True)
10
 
11
  #device = "cuda" if torch.cuda.is_available() else "cpu"
12
- model = AutoModelForCausalLM.from_pretrained('gokaygokay/Florence-2-SD3-Captioner', trust_remote_code=True).to("cuda:0").eval()
13
 
14
- processor = AutoProcessor.from_pretrained('gokaygokay/Florence-2-SD3-Captioner', trust_remote_code=True)
15
 
16
 
17
  TITLE = "# Florence-2-SD3-Captioner"
 
9
  subprocess.run('pip install einops', shell=True)
10
 
11
  #device = "cuda" if torch.cuda.is_available() else "cpu"
12
+ model = AutoModelForCausalLM.from_pretrained('gokaygokay/Florence-2-Flux-Large', trust_remote_code=True).to("cuda:0").eval()
13
 
14
+ processor = AutoProcessor.from_pretrained('gokaygokay/Florence-2-Flux-Large', trust_remote_code=True)
15
 
16
 
17
  TITLE = "# Florence-2-SD3-Captioner"