Update app.py

app.py

@@ -177,7 +177,7 @@ def load_model_from_s3(key: Optional[str] = None):
     return model
 
 
-def load_baseline_S3(bucket: str, key: str):
+def load_baseline_S3():
     bucket = os.getenv('S3_BUCKET')
     BASELINE_KEY = "mlflow/models/baseline_c87008d290ec435eb93cc302e41ce934.pkl"
     response = s3.get_object(Bucket=bucket, Key=BASELINE_KEY)