Spaces:

SwapnaneelBanerjee
/

smart_farming

Running

SwapnaneelBanerjee commited on Feb 12

Commit

a757943

•

1 Parent(s): 0c3b629

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 extractor = AutoFeatureExtractor.from_pretrained("susnato/plant_disease_detection-beans")
 model = AutoModelForImageClassification.from_pretrained("susnato/plant_disease_detection-beans")
-labels = ['angular_leaf_spot', 'rust', 'healthy']
 def classify(im):
   features = extractor(im, return_tensors='pt')

 extractor = AutoFeatureExtractor.from_pretrained("susnato/plant_disease_detection-beans")
 model = AutoModelForImageClassification.from_pretrained("susnato/plant_disease_detection-beans")
+labels = ['angular leaf spot', 'rust', 'healthy']
 def classify(im):
   features = extractor(im, return_tensors='pt')