ejschwartz commited on
Commit
d780e7e
·
1 Parent(s): ddcc7db
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +1 -0
README.md CHANGED
@@ -10,4 +10,4 @@ pinned: false
10
  short_description: Nova 6.7B Space
11
  ---
12
 
13
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
10
  short_description: Nova 6.7B Space
11
  ---
12
 
13
+ This is a space for testing the [Nova 6.7B binary code recovery model](https://huggingface.co/lt-asset/nova-6.7b-bcr). For more information about the model, please refer to the [model card](https://huggingface.co/lt-asset/nova-6.7b-bcr).
app.py CHANGED
@@ -17,6 +17,7 @@ nova_tokenizer = NovaTokenizer(tokenizer)
17
  model = NovaForCausalLM.from_pretrained('lt-asset/nova-6.7b-bcr', torch_dtype=torch.bfloat16).eval()
18
 
19
  examples = json.load(open("humaneval_decompile_nova_6.7b.json", "r"))
 
20
 
21
  @spaces.GPU
22
  def predict(type, normalized_asm):
 
17
  model = NovaForCausalLM.from_pretrained('lt-asset/nova-6.7b-bcr', torch_dtype=torch.bfloat16).eval()
18
 
19
  examples = json.load(open("humaneval_decompile_nova_6.7b.json", "r"))
20
+ print(f"Example {examples[0]}")
21
 
22
  @spaces.GPU
23
  def predict(type, normalized_asm):