Merge pull request #223 from koxudaxi/fix_model_file

Fix model file
koxudaxi · Dec 3, 2021 · 127dddd · 127dddd
2 parents d97a3a2 + 8ff4b62
commit 127dddd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fastapi_code_generator/__main__.py b/fastapi_code_generator/__main__.py
@@ -30,6 +30,8 @@ def main(
     input_text: str = input_file.read()
     if model_file:
         model_path = Path(model_file).with_suffix('.py')
+    else:
+        model_path = MODEL_PATH
     return generate_code(input_name, input_text, output_dir, template_dir, model_path)