Chandranshu Jain commited on
Commit
c7e73d4
1 Parent(s): ace96db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,13 +83,13 @@ def get_conversational_chain():
83
  #repo_id='meta-llama/Meta-Llama-3-70B'
84
  repo_id = 'microsoft/Phi-3-mini-4k-instruct'
85
  llm = HuggingFaceEndpoint(
86
- repo_id=repo_id, temperature=0.3)
87
  #tokenizer = AutoTokenizer.from_pretrained("google/gemma-1.1-2b-it")
88
  #llm = AutoModelForCausalLM.from_pretrained("google/gemma-1.1-2b-it")
89
 
90
  #llm = pipeline("text-generation", model="google/gemma-1.1-2b-it")
91
 
92
- pt = ChatPromptTemplate.from_template(template)
93
  # Retrieve and generate using the relevant snippets of the blog.
94
  retriever = db.as_retriever()
95
  rag_chain = (
 
83
  #repo_id='meta-llama/Meta-Llama-3-70B'
84
  repo_id = 'microsoft/Phi-3-mini-4k-instruct'
85
  llm = HuggingFaceEndpoint(
86
+ repo_id=repo_id, max_length=512, temperature=0.5, token=userdata.get('HUGGING_FACE_API_KEY'))
87
  #tokenizer = AutoTokenizer.from_pretrained("google/gemma-1.1-2b-it")
88
  #llm = AutoModelForCausalLM.from_pretrained("google/gemma-1.1-2b-it")
89
 
90
  #llm = pipeline("text-generation", model="google/gemma-1.1-2b-it")
91
 
92
+ pt = ChatPromptTemplate.from_template(prompt_template)
93
  # Retrieve and generate using the relevant snippets of the blog.
94
  retriever = db.as_retriever()
95
  rag_chain = (