Spaces:
Running
Running
update
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ from sklearn.metrics.pairwise import cosine_similarity
|
|
| 14 |
from sklearn.preprocessing import normalize
|
| 15 |
|
| 16 |
st.set_page_config(
|
| 17 |
-
page_title="Image
|
| 18 |
layout="wide",
|
| 19 |
initial_sidebar_state="expanded"
|
| 20 |
)
|
|
@@ -43,11 +43,11 @@ def load_local_models():
|
|
| 43 |
gc.collect()
|
| 44 |
|
| 45 |
florence_processor = AutoProcessor.from_pretrained(
|
| 46 |
-
"microsoft/Florence-2-
|
| 47 |
trust_remote_code=True
|
| 48 |
)
|
| 49 |
florence_model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
-
"microsoft/Florence-2-
|
| 51 |
trust_remote_code=True,
|
| 52 |
torch_dtype=torch.float32
|
| 53 |
)
|
|
|
|
| 14 |
from sklearn.preprocessing import normalize
|
| 15 |
|
| 16 |
st.set_page_config(
|
| 17 |
+
page_title="Image Captioning Refinement Fusion System",
|
| 18 |
layout="wide",
|
| 19 |
initial_sidebar_state="expanded"
|
| 20 |
)
|
|
|
|
| 43 |
gc.collect()
|
| 44 |
|
| 45 |
florence_processor = AutoProcessor.from_pretrained(
|
| 46 |
+
"microsoft/Florence-2-Base",
|
| 47 |
trust_remote_code=True
|
| 48 |
)
|
| 49 |
florence_model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
+
"microsoft/Florence-2-Base",
|
| 51 |
trust_remote_code=True,
|
| 52 |
torch_dtype=torch.float32
|
| 53 |
)
|