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
|