Tonic commited on
Commit
4df882a
Β·
unverified Β·
1 Parent(s): dca0c5c

fix display bug

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -64,9 +64,8 @@ def load_model():
64
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
65
  device_map="auto" if torch.cuda.is_available() else None,
66
  trust_remote_code=True,
67
- token=hf_token,
68
- use_auth_token=hf_token # Backward compatibility
69
- )
70
 
71
  print("βœ… Osmosis Structure model loaded successfully!")
72
  return True
 
64
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
65
  device_map="auto" if torch.cuda.is_available() else None,
66
  trust_remote_code=True,
67
+ token=hf_token
68
+ )
 
69
 
70
  print("βœ… Osmosis Structure model loaded successfully!")
71
  return True