ejschwartz commited on
Commit
0a4c2da
·
1 Parent(s): 7a59fcd

clarify failure

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ def infer(code):
101
 
102
  field_prompt_result, fields, field_helper_result = field_prompt(code)
103
  if len(fields) == 0:
104
- field_output = "No fields"
105
  else:
106
  field_input_ids = tokenizer.encode(field_prompt_result, return_tensors="pt").cuda()[
107
  :, : 8192 - 1024
 
101
 
102
  field_prompt_result, fields, field_helper_result = field_prompt(code)
103
  if len(fields) == 0:
104
+ field_output = "Failed to parse fields" if field_prompt_result is None else "No fields"
105
  else:
106
  field_input_ids = tokenizer.encode(field_prompt_result, return_tensors="pt").cuda()[
107
  :, : 8192 - 1024