Fix model file

koxudaxi · Dec 2, 2021 · 8ff4b62 · 8ff4b62
1 parent d97a3a2
commit 8ff4b62
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)