Update README.md

README.md

@@ -17,7 +17,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import os
 
-model_path = 'LLM4Binary/llm4decompile-1.3b-v1.5' # V1.5 Model
+model_path = 'Neo111x/falcon3-decompiler-3b' # V1.5 Model
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path,torch_dtype=torch.bfloat16).cuda()
 
@@ -53,7 +53,6 @@ char * func0(char **param_1,int param_2)
 before = f"# This is the assembly code:\n"#prompt
 after = "\n# What is the source code?\n"#prompt
 asm_func = before+asm_func.strip()+after
-model_path = "Neo111x/falcon3-decompiler-3b"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype="auto", device_map="auto").to("cuda:0")