Update app.py

app.py

@@ -9,7 +9,7 @@ app = FastAPI()
 
 # Load the pre-trained model from within the container
 model_uri = "model.pth"
-model = torch.load(model_uri, weights_only=True)
+model = torch.load(model_uri, weights_only=False)
 
 # Define input schema for JSON requests
 class ImageInput(BaseModel):