model67 replaced with best_model.pth

app.py

@@ -11,7 +11,8 @@ class Prediction:
 
 app = fastapi.FastAPI(docs_url="/")
 
-model = torch.load("model67.bin", map_location='cpu')
+# model = torch.load("model67.bin", map_location='cpu')
+model = torch.load("best_model.pth", map_location='cpu')
 print(model)
 # Define a function to preprocess the input image
 def preprocess_input(input: fastapi.UploadFile):