Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ from langchain.prompts import PromptTemplate
6
  # Initialize the chatbot
7
  HF_TOKEN = os.getenv("HF_TOKEN")
8
  llm = HuggingFaceEndpoint(
9
- #repo_id="google/gemma-1.1-7b-it", # Update this to your desired model
10
- repo_id="meta-llama/Meta-Llama-3.1-8B",
11
  task="text-generation",
12
  max_new_tokens=512,
13
  top_k=5,
 
6
  # Initialize the chatbot
7
  HF_TOKEN = os.getenv("HF_TOKEN")
8
  llm = HuggingFaceEndpoint(
9
+ repo_id="google/gemma-1.1-7b-it", # Update this to your desired model
10
+ #repo_id="meta-llama/Meta-Llama-3.1-8B",
11
  task="text-generation",
12
  max_new_tokens=512,
13
  top_k=5,