Update app.py (#14)

- Update app.py (7d8f3c55175cc1b1a17606f072a4163e84ae4522)


Co-authored-by: SIDHARTHA D <dsid271@users.noreply.huggingface.co>

app.py

@@ -1,3 +1,7 @@
+import os
+BACKEND = 'jax' # You can use any backend here
+os.environ['KERAS_BACKEND'] = BACKEND
+
 # app.py (or main.py)
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel