Xalphinions commited on
Commit
86ab2cb
·
verified ·
1 Parent(s): 08093ea

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ if __name__ == "__main__":
43
  import argparse
44
 
45
  parser = argparse.ArgumentParser(description="Watermelon sweetness predictor")
46
- parser.add_argument("--model_path", type=str, required=True, help="Path to the trained model")
47
  args = parser.parse_args()
48
 
49
  model = load_model(args.model_path)
 
43
  import argparse
44
 
45
  parser = argparse.ArgumentParser(description="Watermelon sweetness predictor")
46
+ parser.add_argument("--model_path", type=str, default="./models/model_15_20250405-033557.pt", help="Path to the trained model")
47
  args = parser.parse_args()
48
 
49
  model = load_model(args.model_path)