sridharKikkeri commited on
Commit
16d5059
·
verified ·
1 Parent(s): 75c45d2

Update GetXpath.py

Browse files
Files changed (1) hide show
  1. GetXpath.py +1 -1
GetXpath.py CHANGED
@@ -7,7 +7,7 @@ model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b")
7
 
8
  def generate(prompt):
9
  inputs = tokenizer(prompt, return_tensors="pt")
10
- outputs = model.generate(inputs['input_ids'], max_length=100)
11
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
12
  return response
13
 
 
7
 
8
  def generate(prompt):
9
  inputs = tokenizer(prompt, return_tensors="pt")
10
+ outputs = model.generate(inputs['input_ids'], max_length=100000)
11
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
12
  return response
13