Update app.py

app.py

@@ -23,7 +23,24 @@ def generate(text: str):
 
 @app.post("/uploadfile/")
 async def create_upload_file(file: UploadFile = File(...)):
+    # Read the uploaded file content
     request_object_content = await file.read()
-    img = Image.open(io.BytesIO(request_object_content))
-    pred, idx, probs = learn.predict(img)
+    
+    try:
+        # Attempt to open the image
+        img = Image.open(io.BytesIO(request_object_content))
+    except Exception as e:
+        return {"error": "Failed to open the image file. Make sure it is a valid image file."}
+
+    # Check if img is None or not
+    if img is None:
+        return {"error": "Failed to open the image file."}
+
+    try:
+        # Assuming 'learn' is your image classifier model
+        pred, idx, probs = learn.predict(img)
+    except Exception as e:
+        return {"error": "Failed to make predictions."}
+
+    # Assuming categories is a list of category labels
     return dict(zip(categories, map(float, probs)))