Spaces:

Afsha001
/

Image_captioning

Running

Afsha001 commited on 5 days ago

Commit

f2e1a00

verified ·

1 Parent(s): 3149f37

update florence

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,11 +43,11 @@ def load_local_models():
     gc.collect()
     florence_processor = AutoProcessor.from_pretrained(
-        "microsoft/Florence-2-Base",
         trust_remote_code=True
     )
     florence_model = AutoModelForCausalLM.from_pretrained(
-        "microsoft/Florence-2-Base",
         trust_remote_code=True,
         torch_dtype=torch.float32
     )

     gc.collect()
     florence_processor = AutoProcessor.from_pretrained(
+        "microsoft/Florence-2-large",
         trust_remote_code=True
     )
     florence_model = AutoModelForCausalLM.from_pretrained(
+        "microsoft/Florence-2-large",
         trust_remote_code=True,
         torch_dtype=torch.float32
     )