PinHsuan commited on
Commit
fb34f6c
·
verified ·
1 Parent(s): 13c0689

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
43
  FOLD = 5
44
  MODEL_PATH = f"best_model_fold_{FOLD}.pt"
45
 
 
 
46
 
47
  model = DualStreamTransformer(feat_num_1=24, feat_num_2=10, d_model=32).to(DEVICE)
48
  metric_fc = ArcMarginProduct(32, 2).to(DEVICE)
 
43
  FOLD = 5
44
  MODEL_PATH = f"best_model_fold_{FOLD}.pt"
45
 
46
+ plt.rcParams['font.sans-serif'] = ['Noto Sans CJK TC', 'Droid Sans Fallback', 'Arial Unicode MS']
47
+ plt.rcParams['axes.unicode_minus'] = False
48
 
49
  model = DualStreamTransformer(feat_num_1=24, feat_num_2=10, d_model=32).to(DEVICE)
50
  metric_fc = ArcMarginProduct(32, 2).to(DEVICE)