TharunSivamani commited on
Commit
376ef72
1 Parent(s): 6a33f9a
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -9,12 +9,11 @@ model = AutoModelForCausalLM.from_pretrained(
9
  )
10
  model.config.use_cache = False
11
 
12
- tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
13
- tokenizer.pad_token = tokenizer.eos_token
14
-
15
  adapter_path = 'checkpoint-500'
16
  model.load_adapter(adapter_path)
17
 
 
 
18
 
19
  def generate_context(prompt, tokens=300):
20
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=tokens)
 
9
  )
10
  model.config.use_cache = False
11
 
 
 
 
12
  adapter_path = 'checkpoint-500'
13
  model.load_adapter(adapter_path)
14
 
15
+ tokenizer = AutoTokenizer.from_pretrained("checkpoint-500", trust_remote_code=True)
16
+ tokenizer.pad_token = tokenizer.eos_token
17
 
18
  def generate_context(prompt, tokens=300):
19
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=tokens)