Update app.py

app.py

@@ -1,45 +1,36 @@
-from flask import Flask, request, jsonify, send_file
+from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import Optional
+import tensorflow as tf
 from PIL import Image, ImageOps
 import numpy as np
-import tensorflow as tf
 import io
 
-app = Flask(__name__)
+app = FastAPI()
 
-# Load the TensorFlow model
+# Load your model
 model = tf.keras.models.load_model('my_model2.h5')
 
-def import_and_predict(image_data):
-    image = ImageOps.fit(image_data, (100, 100), Image.ANTIALIAS)
+class ImageData(BaseModel):
+    file: bytes
+
+@app.post("/predict/")
+async def predict(image_data: ImageData):
+    image = Image.open(io.BytesIO(image_data.file))
+    image = ImageOps.fit(image, (100, 100), Image.ANTIALIAS)
     image = image.convert('RGB')
     image = np.asarray(image)
-    image = (image.astype(np.float32) / 255.0)
+    image = image.astype(np.float32) / 255.0
     img_reshape = image[np.newaxis, ...]
-    prediction = model.predict(img_reshape)
-    return prediction
 
-@app.route('/')
-def index():
-    return "Welcome to the Glaucoma Detector API!"
+    prediction = model.predict(img_reshape)
+    pred = prediction[0][0]
 
-@app.route('/predict', methods=['POST'])
-def predict():
-    if 'file' not in request.files:
-        return jsonify({'error': 'No file part'}), 400
-    file = request.files['file']
-    if file.filename == '':
-        return jsonify({'error': 'No selected file'}), 400
-    if file and file.filename.lower().endswith('.jpg'):
-        image = Image.open(io.BytesIO(file.read()))
-        prediction = import_and_predict(image)
-        pred = prediction[0][0]
-        if pred > 0.5:
-            result = "Your eye is Healthy. Great!"
-        else:
-            result = "You are affected by Glaucoma. Please consult an ophthalmologist as soon as possible."
-        return jsonify({'prediction': result})
-    else:
-        return jsonify({'error': 'Invalid file format'}), 400
+    result = {
+        'prediction': 'Healthy' if pred > 0.5 else 'Affected by Glaucoma'
+    }
+    return result
 
-if __name__ == '__main__':
-    app.run(port=5000, debug=True)  # You can specify any port you prefer
+@app.get("/")
+def greet():
+    return {"Hello": "World!"}