Fred808 commited on
Commit
b29fd51
·
verified ·
1 Parent(s): dbe00f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ print("[INFO] Loading Florence-2-base model on CPU...")
35
  try:
36
  processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)
37
  model = AutoModelForCausalLM.from_pretrained(
38
- "microsoft/Florence-2-Large",
39
  trust_remote_code=True,
40
  attn_implementation="eager"
41
  ).to(DEVICE).eval()
 
35
  try:
36
  processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)
37
  model = AutoModelForCausalLM.from_pretrained(
38
+ "microsoft/Florence-2-Base",
39
  trust_remote_code=True,
40
  attn_implementation="eager"
41
  ).to(DEVICE).eval()