weiofj

main.py

@@ -338,7 +338,9 @@ async def predict_single_image_self(file: UploadFile):
 
         # Convert prediction indices to class labels
         predicted_class_index = np.argmax(predictions)
-        predicted_class_label = class_plant_leaves_merged.get(predicted_class_index, "Unknown")
+        predicted_class_label = class_plant_leaves_merged.get(
+            predicted_class_index, "Unknown"
+        )
 
         matched_rows = df[df["Plant Name"] == predicted_class_label.lower()]
         matched_rows_json = matched_rows.to_json(orient="records")
@@ -411,5 +413,9 @@ async def predict_single_image_self_demo(file: UploadFile):
         return {"error": str(e)}
 
 
+def hi():
+    return "si"
+
+
 if __name__ == "__main__":
     uvicorn.run(app, port=8000)