Update app.py

app.py

@@ -5,13 +5,10 @@ from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 
-# Allowlist the required global for PyTorch model loading
-torch.serialization.add_safe_globals(['cloudpickle.cloudpickle._make_skeleton_class'])
-
 app = FastAPI()
 
 # Load the pre-trained model from within the container
-model_uri = "/app/model.pth"
+model_uri = "model.pth"
 model = torch.load(model_uri, weights_only=True)
 
 # Define input schema for JSON requests