Update app.py
This commit is contained in:
parent
b860d5b90a
commit
2bbb35a0d8
1
app.py
1
app.py
|
@ -17,7 +17,6 @@ base_model = "decapoda-research/llama-7b-hf"
|
||||||
adapter_model = "project-baize/baize-lora-7B"
|
adapter_model = "project-baize/baize-lora-7B"
|
||||||
tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
|
tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
|
||||||
|
|
||||||
global total_count
|
|
||||||
total_count = 0
|
total_count = 0
|
||||||
def predict(text,
|
def predict(text,
|
||||||
chatbot,
|
chatbot,
|
||||||
|
|
Loading…
Reference in New Issue