Update main.py

main.py

@@ -15,7 +15,7 @@ async def api_health_check():
 
 @app.get("/download/{plant_name}")
 async def download_model(plant_name: str):
-    filename = f"{plant_name}_model.keras"
+    filename = f"model_{plant_name}.tflite"
     file_path = os.path.join(MODEL_DIRECTORY, filename)
 
     # Check if file exists