Update app.py
All checks were successful
society-ai-hub-container-cache Actions Demo / build (push) Successful in 28s

This commit is contained in:
badhezi 2024-12-11 08:21:19 +00:00
parent 11f48f47e3
commit 3cfb8d5fb4

2
app.py

@ -12,7 +12,7 @@ from langchain.prompts.chat import (
llm = ChatOpenAI( llm = ChatOpenAI(
temperature=0.7, temperature=0.7,
model_name="llama-3.2-3B-instruct", # Replace with your specific model or endpoint if required model_name="llama-3.2-3B-instruct", # Replace with your specific model or endpoint if required
openai_api_base="https://hub.societyai.com/models/llama-3-2-3b-test/openai/v1", openai_api_base="https://hub.societyai.com/models/llama-3-2-3b/openai/v1",
) )
# Set up the memory for the chatbot # Set up the memory for the chatbot