Hugging Face's logo

Spaces:
brendenc
/
Street_Segmentation
Runtime error

App Files Community
brendenc commited on
Commit
e929201
•
1 Parent(s): 5a226a5

Updated from colab

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import matplotlib.pyplot as plt
5
 
6
 
7
- feature_extractor = AutoFeatureExtractor.from_pretrained("brendenc/my-segmentation-model")
8
  model = AutoModelForImageClassification.from_pretrained("brendenc/my-segmentation-model")
9
 
10
  def classify(im):
 
4
  import matplotlib.pyplot as plt
5
 
6
 
7
+ extractor = AutoFeatureExtractor.from_pretrained("brendenc/my-segmentation-model")
8
  model = AutoModelForImageClassification.from_pretrained("brendenc/my-segmentation-model")
9
 
10
  def classify(im):