diff --git a/main.py b/main.py index a6b8eb8..56b9f6b 100644 --- a/main.py +++ b/main.py @@ -68,7 +68,9 @@ def upload(): import traceback; traceback.print_exc() return Response("Unknown error", 400) - grass_touching_similarity = get_grass_touching_similarity(url_for('uploads', filename=f"{image_uuid}.{image_type}")) + print(request.url_root.rstrip('/') + url_for('uploads', filename=f"{image_uuid}.{image_type}")) + + grass_touching_similarity = get_grass_touching_similarity(request.url_root.rstrip('/') + url_for('uploads', filename=f"{image_uuid}.{image_type}")) if not grass_touching_similarity >= 0.8: return Response(f"Image not touching grass. Cosine similarity: {grass_touching_similarity}", 401)