JustinLin610 commited on
Commit
3452998
1 Parent(s): 6632346
Files changed (2) hide show
  1. app.py +2 -2
  2. notebooks/caption_infer.ipynb +0 -0
app.py CHANGED
@@ -11,8 +11,8 @@ import torch
11
  import numpy as np
12
  from fairseq import utils, tasks
13
  from fairseq import checkpoint_utils
14
- from OFA.utils.eval_utils import eval_step
15
- from OFA.tasks.mm_tasks.caption import CaptionTask
16
  from models.ofa import OFAModel
17
  from PIL import Image
18
  from torchvision import transforms
 
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
notebooks/caption_infer.ipynb DELETED
The diff for this file is too large to render. See raw diff