MudassirFayaz commited on
Commit
72c67b3
·
verified ·
1 Parent(s): 69eb5bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -4,8 +4,7 @@ import torch
4
 
5
  # Load the tokenizer and model
6
  tokenizer = AutoTokenizer.from_pretrained('tokenizer-llama_career_0.5.0')
7
- model = AutoModelForCausalLM.from_pretrained('llama-2-7b-career-0.5.0',
8
- adapter_name_or_path='llama-2-7b-career-0.5.0/pytorch_model.bin')
9
 
10
  # Function to generate response
11
  def generate_response(input_text):
 
4
 
5
  # Load the tokenizer and model
6
  tokenizer = AutoTokenizer.from_pretrained('tokenizer-llama_career_0.5.0')
7
+ model = AutoModelForCausalLM.from_pretrained('llama-2-7b-career-0.5.0','llama-2-7b-career-0.5.0/pytorch_model.bin')
 
8
 
9
  # Function to generate response
10
  def generate_response(input_text):