Upload 3 files

Dockerfile

@@ -0,0 +1,12 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+EXPOSE 7860
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,30 @@
+from fastapi import FastAPI, File, UploadFile
+import tensorflow as tf
+import numpy as np
+from PIL import Image
+import io
+
+app = FastAPI()
+
+# Load model
+model = tf.keras.models.load_model("model/saved_model")
+
+def preprocess(image):
+    image = image.resize((224, 224))
+    image = np.array(image) / 255.0
+    return np.expand_dims(image, axis=0)
+
+@app.get("/")
+def root():
+    return {"message": "Model is running!"}
+
+@app.post("/predict")
+async def predict(file: UploadFile = File(...)):
+    image = Image.open(io.BytesIO(await file.read())).convert("RGB")
+    input_data = preprocess(image)
+    
+    prediction = model.predict(input_data)
+    
+    return {
+        "prediction": prediction.tolist()
+    }