Spaces:

Dhahlan2000
/

banana_ripeness_level_detection

Running

Dhahlan2000 commited on 3 days ago

Commit

0d49fc7

verified ·

1 Parent(s): 49f68da

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ model.eval()
 feature_extractor = ViTFeatureExtractor.from_pretrained('Dhahlan2000/banana_ripeness_level_detection')
 # Class labels
-predicted_classes = ['Overripe', 'ripe', 'rotten', 'unripe']
 # Function for inference
 def classify_fruit(image):

 feature_extractor = ViTFeatureExtractor.from_pretrained('Dhahlan2000/banana_ripeness_level_detection')
 # Class labels
+predicted_classes = ['Overripe', 'ripe', 'rotten', 'unripe', 'new']
 # Function for inference
 def classify_fruit(image):