Update app.py

This commit is contained in:
HongCheng 2024-05-03 01:02:55 +09:00
parent 683c40df5e
commit b83270bb13

28
app.py
View File

@ -1,16 +1,18 @@
import os
#model = "EmoLLM_aiwei"
# model = "EmoLLM_Model"
model = "Llama3_Model"
os.system('streamlit run web_demo_Llama3.py --server.address=0.0.0.0 --server.port 7860')
if model == "EmoLLM_aiwei":
os.system("python download_model.py ajupyter/EmoLLM_aiwei")
os.system('streamlit run web_demo-aiwei.py --server.address=0.0.0.0 --server.port 7860')
elif model == "EmoLLM_Model":
os.system("python download_model.py jujimeizuo/EmoLLM_Model")
os.system('streamlit run web_internlm2.py --server.address=0.0.0.0 --server.port 7860')
elif model == "Llama3_Model":
os.system('streamlit run web_demo_Llama3.py --server.address=0.0.0.0 --server.port 7968')
else:
print("Please select one model")
# #model = "EmoLLM_aiwei"
# # model = "EmoLLM_Model"
# model = "Llama3_Model"
# if model == "EmoLLM_aiwei":
# os.system("python download_model.py ajupyter/EmoLLM_aiwei")
# os.system('streamlit run web_demo-aiwei.py --server.address=0.0.0.0 --server.port 7860')
# elif model == "EmoLLM_Model":
# os.system("python download_model.py jujimeizuo/EmoLLM_Model")
# os.system('streamlit run web_internlm2.py --server.address=0.0.0.0 --server.port 7860')
# elif model == "Llama3_Model":
# os.system('streamlit run web_demo_Llama3.py --server.address=0.0.0.0 --server.port 7860')
# else:
# print("Please select one model")