update download_model

This commit is contained in:
HongCheng 2024-04-22 09:04:11 +09:00
parent 7851977e13
commit 5d0f60478b
2 changed files with 2 additions and 1 deletions

1
app.py
View File

@ -12,6 +12,7 @@ elif model == "EmoLLM_Model":
os.system('streamlit run web_internlm2.py --server.address=0.0.0.0 --server.port 7860')
elif model == "Llama3_Model":
os.system("python download_model.py chg0901/EmoLLM-Llama3-8B-Instruct2.0")
# os.system('streamlit run web_demo-Llama3_online.py --server.address=0.0.0.0 --server.port 7860')
os.system('streamlit run web_demo-Llama3_online.py --server.address=0.0.0.0 --server.port 7860')
else:
print("Please select one model")

View File

@ -58,6 +58,6 @@ os.rmdir(temp_dir)
os.remove(output_filename)
download(model_repo='jujimeizuo/EmoLLM_Model', output='model')
download(model_repo=model_repo, output='model')
print("Model bin file download complete")