Update app.py

app.py

@@ -36,6 +36,26 @@ from rife_model import load_rife_model, rife_inference_with_latents
 from huggingface_hub import hf_hub_download, snapshot_download
 import gc
 
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import FileResponse
+from huggingface_hub import hf_hub_download, snapshot_download
+
+from data_class.GenerateResponse import GenerateResponse
+from data_class.GenerateRequest import GenerateRequest
+
+app = FastAPI()
+
+# CORS Configuration
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")
@@ -486,6 +506,12 @@ with gr.Blocks() as demo:
     enhance_button.click(enhance_prompt_func, inputs=[prompt], outputs=[prompt])
     video_input.upload(resize_if_unfit, inputs=[video_input], outputs=[video_input])
 
+@app.get("/health")
+def health_check():
+    """Simple health check endpoint"""
+    return {"status": "OK"}
+
+
 if __name__ == "__main__":
     demo.queue(max_size=15)
     demo.launch()