sharktide commited on
Commit
c9ba775
·
verified ·
1 Parent(s): 3d2a6b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -114,7 +114,7 @@ async def predict_fruitbot0(file: UploadFile = File(...)):
114
  prediction1 = fruitbot_expanded.predict(img_array) # Get predictions
115
 
116
  predicted_class_idx = np.argmax(prediction1, axis=1)[0] # Get predicted class index
117
- predicted_class = FRUITBOT_EXPANDED_CLASSESCLASSES[predicted_class_idx] # Convert to class name
118
 
119
  return JSONResponse(content={"prediction": predicted_class})
120
 
 
114
  prediction1 = fruitbot_expanded.predict(img_array) # Get predictions
115
 
116
  predicted_class_idx = np.argmax(prediction1, axis=1)[0] # Get predicted class index
117
+ predicted_class = FRUITBOT_EXPANDED_CLASSES[predicted_class_idx] # Convert to class name
118
 
119
  return JSONResponse(content={"prediction": predicted_class})
120