prithivMLmods commited on
Commit
ae45ee1
·
verified ·
1 Parent(s): f4268c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -76,10 +76,10 @@ css = """
76
  # Ensure you have enough VRAM/RAM available. `device_map="auto"` will help distribute them.
77
 
78
  MODEL_IDS = {
79
- "Florence-2-base": "microsoft/Florence-2-base",
80
- "Florence-2-base-ft": "microsoft/Florence-2-base-ft",
81
- "Florence-2-large": "microsoft/Florence-2-large",
82
- "Florence-2-large-ft": "microsoft/Florence-2-large-ft",
83
  }
84
 
85
  models = {}
 
76
  # Ensure you have enough VRAM/RAM available. `device_map="auto"` will help distribute them.
77
 
78
  MODEL_IDS = {
79
+ "Florence-2-base": "florence-community/Florence-2-base",
80
+ "Florence-2-base-ft": "florence-community/Florence-2-base-ft",
81
+ "Florence-2-large": "florence-community/Florence-2-large",
82
+ "Florence-2-large-ft": "florence-community/Florence-2-large-ft",
83
  }
84
 
85
  models = {}