gravelcompbio commited on
Commit
4bd4a2a
1 Parent(s): 0455452

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import numpy as np
8
  import pandas as pd
9
  import matplotlib.pyplot as plt
10
 
11
- #import torch
12
 
13
  from tokenization_esm import EsmTokenizer
14
  from modeling_esm import EsmForSequenceClassificationMHACustom
 
8
  import pandas as pd
9
  import matplotlib.pyplot as plt
10
 
11
+ import torch, torchvision
12
 
13
  from tokenization_esm import EsmTokenizer
14
  from modeling_esm import EsmForSequenceClassificationMHACustom