SwapnaneelBanerjee commited on
Commit
a757943
1 Parent(s): 0c3b629

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
6
  extractor = AutoFeatureExtractor.from_pretrained("susnato/plant_disease_detection-beans")
7
  model = AutoModelForImageClassification.from_pretrained("susnato/plant_disease_detection-beans")
8
 
9
- labels = ['angular_leaf_spot', 'rust', 'healthy']
10
 
11
  def classify(im):
12
  features = extractor(im, return_tensors='pt')
 
6
  extractor = AutoFeatureExtractor.from_pretrained("susnato/plant_disease_detection-beans")
7
  model = AutoModelForImageClassification.from_pretrained("susnato/plant_disease_detection-beans")
8
 
9
+ labels = ['angular leaf spot', 'rust', 'healthy']
10
 
11
  def classify(im):
12
  features = extractor(im, return_tensors='pt')