DB2323 commited on
Commit
d307df7
·
1 Parent(s): bd79d7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@
15
  #iface.launch()
16
  from transformers import LlamaForCausalLM, LlamaTokenizer
17
 
18
- tokenizer = LlamaTokenizer.from_pretrained("/output/path")
19
- model = LlamaForCausalLM.from_pretrained("/output/path")
20
 
21
  prompt = "Hey, are you conscious? Can you talk to me?"
22
  inputs = tokenizer(prompt, return_tensors="pt")
 
15
  #iface.launch()
16
  from transformers import LlamaForCausalLM, LlamaTokenizer
17
 
18
+ tokenizer = LlamaTokenizer.from_pretrained("models/nomic-ai/gpt4all-13b-snoozy")
19
+ model = LlamaForCausalLM.from_pretrained("models/nomic-ai/gpt4all-13b-snoozy")
20
 
21
  prompt = "Hey, are you conscious? Can you talk to me?"
22
  inputs = tokenizer(prompt, return_tensors="pt")