Spaces:
Running
Running
update
Browse files- .gitignore +1 -0
- app.py +1 -1
.gitignore
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
app_backup.py
|
app.py
CHANGED
@@ -69,7 +69,7 @@ st.markdown("""
|
|
69 |
@st.cache_resource
|
70 |
def load_model():
|
71 |
REVISION = 'refs/pr/6'
|
72 |
-
MODEL_NAME = "
|
73 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
74 |
|
75 |
config_model = AutoConfig.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
|
|
|
69 |
@st.cache_resource
|
70 |
def load_model():
|
71 |
REVISION = 'refs/pr/6'
|
72 |
+
MODEL_NAME = "Anonymous-AC/AD-KD"
|
73 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
74 |
|
75 |
config_model = AutoConfig.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
|