Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
0a4c2da
1
Parent(s):
7a59fcd
clarify failure
Browse files
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
|