Hugging Face's logo

Spaces:
PinHsuan
/
Questionnaire_basedModel
Sleeping

App Files Community
PinHsuan commited on
Commit
e694e57
·
verified ·
1 Parent(s): ffd09d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
40
  FOLD = 5
41
  MODEL_PATH = f"best_model_fold_{FOLD}.pt"
42
 
43
- model = DualStreamTransformer(n_feat1=24, n_feat2=10, d_model=32).to(DEVICE)
44
  metric_fc = ArcMarginProduct(32, 2).to(DEVICE)
45
 
46
  if os.path.exists(MODEL_PATH):
 
40
  FOLD = 5
41
  MODEL_PATH = f"best_model_fold_{FOLD}.pt"
42
 
43
+ model = DualStreamTransformer(feat_num_1=24, feat_num_2=10, d_model=32)
44
  metric_fc = ArcMarginProduct(32, 2).to(DEVICE)
45
 
46
  if os.path.exists(MODEL_PATH):