Spaces:
Runtime error
Runtime error
Commit
·
ff4a90e
1
Parent(s):
8845f7a
update
Browse files
app.py
CHANGED
@@ -1,11 +1,12 @@
|
|
1 |
-
import gradio as gr
|
2 |
import os
|
3 |
-
|
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)
|