Update app.py
Browse files
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-
|
| 37 |
model = AutoModelForCausalLM.from_pretrained(
|
| 38 |
-
"microsoft/Florence-2-
|
| 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()
|