ejschwartz commited on
Commit
0adad70
·
1 Parent(s): e2f9f6a
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -71,10 +71,11 @@ def infer(code):
71
 
72
  varstring = ", ".join([f"`{v}`" for v in vars])
73
 
74
- prompt = f"What are the original name and data types of variables {varstring}?\n```{code}\n```"
75
-
76
  var_name = vars[0]
77
 
 
 
 
78
  prompt = code + var_name + ":"
79
  print(prompt)
80
 
 
71
 
72
  varstring = ", ".join([f"`{v}`" for v in vars])
73
 
 
 
74
  var_name = vars[0]
75
 
76
+ # 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
77
+ prompt = f"What are the original name and data types of variables {varstring}?\n```{code}\n```{var_name}"
78
+
79
  prompt = code + var_name + ":"
80
  print(prompt)
81