Spaces:
Sleeping
Sleeping
Add requirements.txt
Browse files- app.py +0 -2
- requirements.txt +5 -0
app.py
CHANGED
@@ -4,8 +4,6 @@ from PIL import Image
|
|
4 |
from torchvision import transforms
|
5 |
import numpy as np
|
6 |
from matplotlib import pyplot as plt
|
7 |
-
from torch.utils.data import Dataset, DataLoader
|
8 |
-
import evaluate
|
9 |
from torch import nn
|
10 |
from transformers import SegformerForSemanticSegmentation
|
11 |
import sys
|
|
|
4 |
from torchvision import transforms
|
5 |
import numpy as np
|
6 |
from matplotlib import pyplot as plt
|
|
|
|
|
7 |
from torch import nn
|
8 |
from transformers import SegformerForSemanticSegmentation
|
9 |
import sys
|
requirements.txt
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
torch==1.12
|
2 |
+
Pillow==9.3.0
|
3 |
+
torchvision==0.13.1
|
4 |
+
matplotlib
|
5 |
+
transformers==4.30.2
|