phiyodr commited on
Commit
0672bf2
·
1 Parent(s): 8c36ac7

Add requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +0 -2
  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