Update app.py

app.py

@@ -1,3 +1,8 @@
+import os
+
+# Set CUDA_VISIBLE_DEVICES to empty string to disable GPU usage
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
+
 from fastapi import FastAPI
 from asgiref.wsgi import WsgiToAsgi
 from flask import Flask, request, render_template
@@ -99,7 +104,7 @@ def index():
     return render_template('index.html', prediction=predictions_list, errors=errors, record=request.form)
 
 # FastAPI app to mount Flask app
-fastapi_app = FastAPI()
+app = FastAPI()  # Change this line from `fastapi_app` to `app`
 
 # Mount Flask app inside FastAPI using WSGIMiddleware
-fastapi_app.mount("/", WSGIMiddleware(flask_app))
+app.mount("/", WSGIMiddleware(flask_app))  # Keep this line