JamalAG commited on
Commit
3e7bb7b
·
1 Parent(s): 8e89e52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,9 +4,8 @@ from langchain.llms import HuggingFaceHub
4
  # Function to return the response
5
  def generate_answer(query):
6
  llm = HuggingFaceHub(
7
- repo_id="TheBloke/goliath-120b-GGUF",
8
- model_kwargs={"temperature": 0.5, "max_length": 64, "max_new_tokens": 512},
9
- task="text2text-generation"
10
  )
11
  prompt = f"""
12
  You are a helpful AI assistant.
 
4
  # Function to return the response
5
  def generate_answer(query):
6
  llm = HuggingFaceHub(
7
+ repo_id="alpindale/goliath-120b",
8
+ model_kwargs={"temperature": 0.5, "max_length": 64, "max_new_tokens": 512}
 
9
  )
10
  prompt = f"""
11
  You are a helpful AI assistant.