Kathirsci commited on
Commit
59e705d
·
verified ·
1 Parent(s): 2ab0cb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -7,11 +7,8 @@ from langchain.prompts import PromptTemplate
7
  # Initialize the chatbot
8
  HF_TOKEN = os.getenv("HF_TOKEN")
9
  llm = HuggingFaceEndpoint(
10
- #repo_id="google/gemma-2b-it",
11
- #repo_id="abhinand/gemma-2b-tamil",
12
- repo_id="google/gemma-1.1-7b-it",
13
- #repo_id="google/gemma-2b",
14
 
 
15
  task="text-generation",
16
  max_new_tokens=512,
17
  top_k=5,
 
7
  # Initialize the chatbot
8
  HF_TOKEN = os.getenv("HF_TOKEN")
9
  llm = HuggingFaceEndpoint(
 
 
 
 
10
 
11
+ repo_id="mistralai/Mistral-7B-Instruct-v0.3",
12
  task="text-generation",
13
  max_new_tokens=512,
14
  top_k=5,