JustinLin610 commited on
Commit
ff4a90e
1 Parent(s): 8845f7a
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,11 +1,12 @@
1
- import gradio as gr
2
  import os
3
- import torch
4
- import numpy as np
5
 
6
  os.system('git clone https://github.com/pytorch/fairseq.git; cd fairseq;'
7
  'pip install --use-feature=in-tree-build ./; cd ..')
8
 
 
 
 
9
  from fairseq import utils, tasks
10
  from utils import checkpoint_utils
11
  from utils.eval_utils import eval_step
@@ -13,6 +14,8 @@ from tasks.mm_tasks.caption import CaptionTask
13
  from models.ofa import OFAModel
14
  from PIL import Image
15
  from torchvision import transforms
 
 
16
 
17
  # Register caption task
18
  tasks.register_task('caption', CaptionTask)
 
 
1
  import os
2
+
 
3
 
4
  os.system('git clone https://github.com/pytorch/fairseq.git; cd fairseq;'
5
  'pip install --use-feature=in-tree-build ./; cd ..')
6
 
7
+
8
+ import torch
9
+ import numpy as np
10
  from fairseq import utils, tasks
11
  from utils import checkpoint_utils
12
  from utils.eval_utils import eval_step
 
14
  from models.ofa import OFAModel
15
  from PIL import Image
16
  from torchvision import transforms
17
+ import gradio as gr
18
+
19
 
20
  # Register caption task
21
  tasks.register_task('caption', CaptionTask)