ashish-soni08 commited on
Commit
330bca7
·
verified ·
1 Parent(s): ba9b7cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model = AutoModelForImageClassification.from_pretrained("saved_model_files")
15
 
16
  labels = ['angular_leaf_spot', 'bean_rust', 'healthy']
17
 
18
- @spaces.GPU(duration=240)
19
  def classify(image: Image.Image) -> Dict[str, float]:
20
  """
21
  Classify an image of a bean plant leaf into one of several health categories.
 
15
 
16
  labels = ['angular_leaf_spot', 'bean_rust', 'healthy']
17
 
18
+ @spaces.GPU(duration=500)
19
  def classify(image: Image.Image) -> Dict[str, float]:
20
  """
21
  Classify an image of a bean plant leaf into one of several health categories.