Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
|
30 |
import streamlit as st
|
31 |
|
32 |
|
33 |
-
st.write(stream('Count to ten'))
|
34 |
|
35 |
# Load pre-trained LLaMA 2 model and tokenizer
|
36 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map='auto')
|
|
|
30 |
import streamlit as st
|
31 |
|
32 |
|
33 |
+
# st.write(stream('Count to ten'))
|
34 |
|
35 |
# Load pre-trained LLaMA 2 model and tokenizer
|
36 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map='auto')
|