Update api.py

api.py

@@ -210,7 +210,7 @@ async def classify(image: UploadFile = File(...)):
     else:
         return {"error": "No image provided"}
 
-model = YOLO("yolo11m.pt")
+yolomodel = YOLO("yolo11m.pt")
 @app.post("/multiclassify")
 async def classify(image: UploadFile = File(...)):
     if image is not None:
@@ -229,7 +229,7 @@ async def classify(image: UploadFile = File(...)):
             cropped_image = cropped_image.resize((64,64))
             img_array = np.array(cropped_image) / 255.0
             img_array = np.expand_dims(img_array, axis=0)
-            predictions = model.predict(img_array)
+            predictions = yolomodel.predict(img_array)
             predicted_class_idx = np.argmax(predictions)
             predicted_class_idx = int(predicted_class_idx)