File size: 566 Bytes
1792117 5f87533 7f51a5a 2856ccc 44131c9 56e7c58 fc6f1e0 b61717e 56e7c58 9861a8a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from transformers import AutoModelForCausalLM, AutoTokenizer
from huggingface_hub import login
import streamlit as st
st.set_page_config(
page_title="Code Generation",
page_icon="🤖",
layout="wide",
initial_sidebar_state="expanded",
)
login(token='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', add_to_git_credential=True)
st.title("Code Generation")
st.write('MODEL: TinyPixel/red1xe/Llama-2-7B-codeGPT')
model_name='lmsys/vicuna-7b-v1.1'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model= AutoModelForCausalLM.from_pretrained(model_name)
|