Upload folder using huggingface_hub
Browse files
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,
|
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 |
|