Caslow commited on
Commit
58e04b1
1 Parent(s): 1903078

add model path in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import inference
4
  def translate_fortran_to_rust(fortran_code, fortran_explain):
5
  """Translate Fortran code to Rust using the provided model."""
6
  # Translation logic, with example stubbed function call
7
- rust_code = inference.main(USER_INPUT_CODE = fortran_code, USER_INPUT_EXPLANATION = fortran_explain)
8
  return rust_code
9
 
10
  default_codes = """
 
4
  def translate_fortran_to_rust(fortran_code, fortran_explain):
5
  """Translate Fortran code to Rust using the provided model."""
6
  # Translation logic, with example stubbed function call
7
+ rust_code = inference.main(USER_INPUT_CODE = fortran_code, USER_INPUT_EXPLANATION = fortran_explain, MODEL_PATH = "lora_model")
8
  return rust_code
9
 
10
  default_codes = """