ejschwartz commited on
Commit
77bd6a8
·
verified ·
1 Parent(s): 2c55bb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
8
  model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.bfloat16).cuda()
9
 
10
  @spaces.GPU
11
- def predict(asm):
12
 
13
  before = f"# This is the assembly code:\n"#prompt
14
  after = "\n# What is the source code?\n"#prompt
 
8
  model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.bfloat16).cuda()
9
 
10
  @spaces.GPU
11
+ def predict(input_asm):
12
 
13
  before = f"# This is the assembly code:\n"#prompt
14
  after = "\n# What is the source code?\n"#prompt