Chandranshu Jain commited on
Commit
befd304
1 Parent(s): cf3ac3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,7 +76,8 @@ def get_conversational_chain():
76
  Answer:
77
  """
78
  #model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3, google_api_key=GOOGLE_API_KEY)
79
- repo_id='meta-llama/Meta-Llama-3-70B'
 
80
  llm = HuggingFaceEndpoint(
81
  repo_id=repo_id, max_length=512, temperature=0.5, token=HUGGING_FACE_API_KEY)
82
  pt = ChatPromptTemplate.from_template(template)
 
76
  Answer:
77
  """
78
  #model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3, google_api_key=GOOGLE_API_KEY)
79
+ repo_id ='google/gemma-1.1-2b-it'
80
+ #repo_id='meta-llama/Meta-Llama-3-70B'
81
  llm = HuggingFaceEndpoint(
82
  repo_id=repo_id, max_length=512, temperature=0.5, token=HUGGING_FACE_API_KEY)
83
  pt = ChatPromptTemplate.from_template(template)