MudassirFayaz commited on
Commit
49bdeaa
·
verified ·
1 Parent(s): 72c67b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +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','llama-2-7b-career-0.5.0/pytorch_model.bin')
8
 
9
  # Function to generate response
10
  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')
8
 
9
  # Function to generate response
10
  def generate_response(input_text):