Spaces:

Anonymous-AC
/

AG-KD-anonymous-Demo

Running

RioJune commited on 26 days ago

Commit

4756a48

1 Parent(s): 78b1606

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -70,7 +70,7 @@ st.markdown("""
 def load_model():
     REVISION = 'refs/pr/6'
     # MODEL_NAME = "RioJune/AD-KD-MICCAI25"
-    MODEL_NAME = '/u/home/lj0/Checkpoints/AD-KD-MICCAI25'
     DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     config_model = AutoConfig.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)

 def load_model():
     REVISION = 'refs/pr/6'
     # MODEL_NAME = "RioJune/AD-KD-MICCAI25"
+    MODEL_NAME = 'Anonymous-AC/AD-KD'
     DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     config_model = AutoConfig.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)