SHAMIL SHAHBAZ AWAN commited on
Commit
044e524
·
verified ·
1 Parent(s): 17c52c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ if not hf_api_key:
32
  else:
33
  # Initialize the model and tokenizer using the API key
34
  try:
35
- model_name = "llama3-70b-8192" # Replace with the correct model name if needed
36
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_api_key)
37
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_api_key)
38
  llm_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
32
  else:
33
  # Initialize the model and tokenizer using the API key
34
  try:
35
+ model_name = "llama-3.1-8b-instant" # Replace with the correct model name if needed
36
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_api_key)
37
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_api_key)
38
  llm_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)