Prajjwalng commited on
Commit
f8baa51
·
verified ·
1 Parent(s): fe6193d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def load_model(model_id):
35
  tokenizer = AutoTokenizer.from_pretrained(model_id, add_eos_token=True)
36
  return base_model, tokenizer
37
 
38
- merged_model, tokenizer = load_model(model_id, adapter_id)
39
 
40
  # Function to generate chatbot response using the provided template
41
  def get_completion(query: str, model, tokenizer) -> str:
 
35
  tokenizer = AutoTokenizer.from_pretrained(model_id, add_eos_token=True)
36
  return base_model, tokenizer
37
 
38
+ merged_model, tokenizer = load_model(model_id)
39
 
40
  # Function to generate chatbot response using the provided template
41
  def get_completion(query: str, model, tokenizer) -> str: