add app

app.py

@@ -0,0 +1,35 @@
+from fastapi import FastAPI, File, UploadFile
+from fastapi.middleware.cors import CORSMiddleware
+import io, base64
+from PIL import Image
+
+from model import predict_galaxy
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.post("/predict")
+async def predict(file: UploadFile = File(...)):
+    image = Image.open(io.BytesIO(await file.read())).convert("RGB")
+
+    overlay, result_text = predict_galaxy(image)
+
+    buf = io.BytesIO()
+    overlay.save(buf, format="PNG")
+
+    predicted_class = result_text.split("\n")[0].split(": ")[1]
+    probability = float(
+        result_text.split("\n")[1].split(": ")[1].replace("%", "")
+    ) / 100
+
+    return {
+        "class": predicted_class,
+        "probability": probability,
+        "heatmap": base64.b64encode(buf.getvalue()).decode()
+    }

robust_galaxy_model .pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b6217b4ef7679ccf90ab16c733ac4fe7810376c389330a7fe663f718114f8823
-size 44790923