zbing commited on
Commit
14bdc95
1 Parent(s): 6908eec

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -11,7 +11,7 @@ app = Flask(__name__)
11
 
12
  # Parse command line arguments
13
  parser = argparse.ArgumentParser(description='Start the Flask server with specified model and device.')
14
- parser.add_argument('--model-path', type=str, required=True, help='Path to the pretrained model')
15
  parser.add_argument('--device', type=str, choices=['cpu', 'gpu'], default='auto', help='Device to use: "cpu", "gpu", or "auto"')
16
  args = parser.parse_args()
17
 
 
11
 
12
  # Parse command line arguments
13
  parser = argparse.ArgumentParser(description='Start the Flask server with specified model and device.')
14
+ parser.add_argument('--model-path', type=str, default="models/Florence-2-base-ft", help='Path to the pretrained model')
15
  parser.add_argument('--device', type=str, choices=['cpu', 'gpu'], default='auto', help='Device to use: "cpu", "gpu", or "auto"')
16
  args = parser.parse_args()
17