diff --git a/app.py b/app.py index dd75608..13458bd 100644 --- a/app.py +++ b/app.py @@ -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") \ No newline at end of file diff --git a/download_model.py b/download_model.py index 9079654..93b18d9 100644 --- a/download_model.py +++ b/download_model.py @@ -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") \ No newline at end of file