Update app.py

app.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, JSONResponse
 import numpy as np
 from tensorflow.keras.models import load_model
 from PIL import Image
@@ -44,6 +44,11 @@ def preprocess_sketch(image_bytes):
     except Exception as e:
         raise ValueError(f"Image processing failed: {str(e)}")
 
+@app.get("/")
+async def health_check():
+    """Health check endpoint"""
+    return JSONResponse(content={"status": "API is running", "model": "GAN Image Generator"})
+
 @app.post("/generate-from-sketch")
 async def generate_from_sketch(file: UploadFile = File(...)):
     try:
@@ -69,3 +74,4 @@ async def generate_from_sketch(file: UploadFile = File(...)):
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
 
+