remove field decoding

app.py

@@ -28,6 +28,7 @@ and I reused some of their own code to do this.
 
 ## Todo
 
+* Add field decoding
 * Add examples
 
 """
@@ -76,11 +77,11 @@ def infer(code):
     var_name = vars[0]
 
     # ejs: Yeah, this var_name thing is really bizarre. But look at https://github.com/lt-asset/resym/blob/main/training_src/fielddecoder_inf.py
-    prompt = f"What are the original name and data types of variables {varstring}?\n```\n{code}\n```{var_name}"
+    var_prompt = f"What are the original name and data types of variables {varstring}?\n```\n{code}\n```{var_name}"
 
-    print(f"Prompt:\n{prompt}")
+    print(f"Prompt:\n{var_prompt}")
 
-    input_ids = tokenizer.encode(prompt, return_tensors="pt").cuda()[:, : 8192 - 1024]
+    input_ids = tokenizer.encode(var_prompt, return_tensors="pt").cuda()[:, : 8192 - 1024]
     var_output = vardecoder_model.generate(
         input_ids=input_ids,
         max_new_tokens=1024,
@@ -114,7 +115,7 @@ def infer(code):
 
     var_output = var_name + ":" + var_output
     field_output = var_name + ":" + field_output
-    return var_output, field_output, varstring
+    return var_output, varstring
 
 
 demo = gr.Interface(
@@ -124,7 +125,7 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Text(label="Var Decoder Output"),
-        gr.Text(label="Field Decoder Output"),
+        #gr.Text(label="Field Decoder Output"),
         gr.Text(label="Generated Variable List"),
     ],
     description=description,