Hugging Face's logo

Spaces:
kuyesu22
/
sunbird-ug
Runtime error

App Files Community
kuyesu22 commited on
Commit
d5bd6e0
·
verified ·
1 Parent(s): a3f750a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ login(token=access_token)
11
 
12
 
13
  # Load model and tokenizer from the Hugging Face Hub
14
- model_name = "kuyesu22/sunbird-ug-lang-v1.0"
15
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
16
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
17
  model = PeftModel.from_pretrained(model, model_name)
 
11
 
12
 
13
  # Load model and tokenizer from the Hugging Face Hub
14
+ model_name = "kuyesu22/sunbird-ug-lang-v1.0-llama-2-7b-hf-lora"
15
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
16
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
17
  model = PeftModel.from_pretrained(model, model_name)