JustinLin610 commited on
Commit
7438ed6
1 Parent(s): 6544c3a
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,6 +4,7 @@ import os
4
  os.system('git clone https://github.com/pytorch/fairseq.git; cd fairseq;'
5
  'git checkout tags/v0.10.1;'
6
  'pip install --use-feature=in-tree-build ./; cd ..')
 
7
  os.system('ls -l')
8
 
9
 
@@ -11,8 +12,8 @@ import torch
11
  import numpy as np
12
  from fairseq import utils, tasks
13
  from fairseq import checkpoint_utils
14
- from utils.eval_utils import eval_step
15
- from tasks.mm_tasks.caption import CaptionTask
16
  from models.ofa import OFAModel
17
  from PIL import Image
18
  from torchvision import transforms
4
  os.system('git clone https://github.com/pytorch/fairseq.git; cd fairseq;'
5
  'git checkout tags/v0.10.1;'
6
  'pip install --use-feature=in-tree-build ./; cd ..')
7
+ os.system('git clone https://github.com/OFA-Sys/OFA.git')
8
  os.system('ls -l')
9
 
10
 
12
  import numpy as np
13
  from fairseq import utils, tasks
14
  from fairseq import checkpoint_utils
15
+ from OFA.utils.eval_utils import eval_step
16
+ from OFA.tasks.mm_tasks.caption import CaptionTask
17
  from models.ofa import OFAModel
18
  from PIL import Image
19
  from torchvision import transforms