Update api.py

api.py

@@ -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 = yolomodel.predict(img_array)
+            predictions = model.predict(img_array)
             predicted_class_idx = np.argmax(predictions)
             predicted_class_idx = int(predicted_class_idx)