Afsha001 commited on
Commit
3149f37
·
verified ·
1 Parent(s): 994d59a
Files changed (1) hide show
  1. app.py +3 -3
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 Caption Fusion System",
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-large",
47
  trust_remote_code=True
48
  )
49
  florence_model = AutoModelForCausalLM.from_pretrained(
50
- "microsoft/Florence-2-large",
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
  )