Update app.py

app.py

@@ -6,8 +6,8 @@ import numpy as np
 from torch.nn import functional as F
 
 
-m = AutoModelForCausalLM.from_pretrained("/mnt/nvme/home/dakota/ckpts/stablelm/7B-sft-combined/checkpoint-8000", torch_dtype=torch.float16).cuda()
-tok = AutoTokenizer.from_pretrained("/mnt/nvme/home/dakota/stablelm_tokenizer")
+m = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-tuned-alpha-7b", torch_dtype=torch.float16).cuda()
+tok = AutoTokenizer.from_pretrained("stabilityai/stablelm-tuned-alpha-7b")
 generator = pipeline('text-generation', model=m, tokenizer=tok, device=0)