JustinLin610 commited on
Commit
653ab88
1 Parent(s): 2f967de

update requirements

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -5
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  import torch
4
  import numpy as np
5
  from fairseq import utils, tasks
6
- from utils import checkpoint_utils
7
  from utils.eval_utils import eval_step
8
  from tasks.mm_tasks.caption import CaptionTask
9
  from models.ofa import OFAModel
 
3
  import torch
4
  import numpy as np
5
  from fairseq import utils, tasks
6
+ from fairseq.utils import checkpoint_utils
7
  from utils.eval_utils import eval_step
8
  from tasks.mm_tasks.caption import CaptionTask
9
  from models.ofa import OFAModel
requirements.txt CHANGED
@@ -1,5 +1 @@
1
- fairseq
2
- ftfy==6.0.3
3
- tensorboardX==2.4.1
4
- pycocotools==2.0.4
5
- pycocoevalcap==1.2
 
1
+ fairseq