Update app.py

This commit is contained in:
Baize 2023-04-04 06:40:33 +00:00 committed by huggingface-web
parent 06a42ee132
commit 7e2a3f2da9
1 changed files with 1 additions and 0 deletions

1
app.py
View File

@ -17,6 +17,7 @@ base_model = "decapoda-research/llama-7b-hf"
adapter_model = "project-baize/baize-lora-7B"
tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
global total_cont
total_cont = 0
def predict(text,
chatbot,