RioJune commited on
Commit
774d5c3
·
1 Parent(s): 826447b
Files changed (2) hide show
  1. .gitignore +1 -0
  2. 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 = "RioJune/AD-KD-MICCAI25"
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)