dd
Browse files- app.py +1 -0
- requirements.txt +2 -1
app.py
CHANGED
@@ -3,6 +3,7 @@ import torch
|
|
3 |
from PIL import Image
|
4 |
from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
|
5 |
from IPython.display import display
|
|
|
6 |
|
7 |
def greet(url):
|
8 |
# load Mask2Former fine-tuned on Cityscapes semantic segmentation
|
|
|
3 |
from PIL import Image
|
4 |
from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
|
5 |
from IPython.display import display
|
6 |
+
import torchvision.transforms as T
|
7 |
|
8 |
def greet(url):
|
9 |
# load Mask2Former fine-tuned on Cityscapes semantic segmentation
|
requirements.txt
CHANGED
@@ -2,4 +2,5 @@ torch
|
|
2 |
transformers
|
3 |
numpy
|
4 |
scipy
|
5 |
-
Image
|
|
|
|
2 |
transformers
|
3 |
numpy
|
4 |
scipy
|
5 |
+
Image
|
6 |
+
torchvision
|