acbdkk commited on
Commit
a564b45
1 Parent(s): d471cce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,10 +1,9 @@
1
  import streamlit as st
2
  import transformers
3
- from transformers import AutoTokenizer
4
  from transformers import LlamaTokenizer
5
  import torch
6
  model = "PY007/TinyLlama-1.1B-Chat-v0.1"
7
- tokenizer = transformers.AutoTokenizer.from_pretrained(model)
8
  pipeline = transformers.pipeline("text-generation",model=model,torch_dtype=torch.bfloat16)
9
 
10
  prompt = "What is 6534+2343?"
 
1
  import streamlit as st
2
  import transformers
 
3
  from transformers import LlamaTokenizer
4
  import torch
5
  model = "PY007/TinyLlama-1.1B-Chat-v0.1"
6
+ tokenizer = transformers.LlamaTokenizer.from_pretrained(model)
7
  pipeline = transformers.pipeline("text-generation",model=model,torch_dtype=torch.bfloat16)
8
 
9
  prompt = "What is 6534+2343?"