dashasabirova commited on
Commit
f705779
1 Parent(s): 7b1729a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import torch
3
  import numpy as np
4
  from transformers import AutoTokenizer, AutoModel
5
  import torch.nn as nn
 
6
 
7
  labels_articles = {1: 'Computer Science', 2: 'Economics', 3: "Electrical Engineering And Systems Science",
8
  4: "Mathematics", 5: "Physics", 6: "Quantitative Biology", 7: "Quantitative Finance",
 
3
  import numpy as np
4
  from transformers import AutoTokenizer, AutoModel
5
  import torch.nn as nn
6
+ from scipy.special import softmax
7
 
8
  labels_articles = {1: 'Computer Science', 2: 'Economics', 3: "Electrical Engineering And Systems Science",
9
  4: "Mathematics", 5: "Physics", 6: "Quantitative Biology", 7: "Quantitative Finance",