Update main.py

main.py

@@ -9,7 +9,7 @@ def predict(decompiled_code):
     with tempfile.TemporaryDirectory() as temp_dir:
 
         code_file_name = os.path.join(temp_dir, "code.decompiled")
-        field_file_name = os.path.join(temp_dir, "code.fields")
+        field_file_name = os.path.join(temp_dir, "code.json")
 
         with open(code_file_name, "w") as f:
             f.write('#include "/home/ReSym/clang-parser/defs.hh"\n' + decompiled_code)
@@ -19,7 +19,8 @@ def predict(decompiled_code):
         print(output)
         print(output.stdout)
 
-        return open(field_file_name).read()
+        field_data = open(field_file_name).read()
+        return field_data
 
 def run():
     demo = gr.Interface(