LLaMA-Factory/src/api_demo.py
hiyouga 471ef69218 update web demo
Former-commit-id: 53d6dc396dbfb1d25236bbc3075404cd5be6b376
2023-08-03 13:28:28 +08:00

15 lines
287 B
Python

import uvicorn
from llmtuner import ChatModel, create_app
def main():
chat_model = ChatModel()
app = create_app(chat_model)
uvicorn.run(app, host="0.0.0.0", port=8000, workers=1)
# Visit http://localhost:8000/docs for document.
if __name__ == "__main__":
main()