Fixes MPS fallback error

app_api.py

@@ -1,4 +1,6 @@
 # app_api.py
+import os
+os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
 
 import base64
 from io import BytesIO
@@ -200,10 +202,6 @@ def inpaint(request: InpaintingRequest):
 @app.get("/status/{job_id}")
 def get_status(job_id: str):
     task = AsyncResult(job_id, app=celery_app)
-    # Check if the task ID exists
-    if task.state == states.PENDING and not task.result:
-        return {"status": "NOT_FOUND"}
-
     if task.state == states.PENDING:
         return {"status": "PENDING"}
     elif task.state == states.STARTED: