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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,9 +33,9 @@ class ImageAnalysisResponse(BaseModel):
33
  # ===== Load Florence-2 Base Model =====
34
  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-Base",
39
  trust_remote_code=True,
40
  attn_implementation="eager"
41
  ).to(DEVICE).eval()
 
33
  # ===== Load Florence-2 Base Model =====
34
  print("[INFO] Loading Florence-2-base model on CPU...")
35
  try:
36
+ processor = AutoProcessor.from_pretrained("microsoft/Florence-2-large", 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()