Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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": "
|
| 80 |
-
"Florence-2-base-ft": "
|
| 81 |
-
"Florence-2-large": "
|
| 82 |
-
"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 = {}
|