lopera47 commited on
Commit
9b5d5d0
·
verified ·
1 Parent(s): d6eb913

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ class BasicAgent: # Some times Inheritance is needed
38
  # )
39
  # Define the LLM
40
  llm = HuggingFaceEndpoint(
41
- endpoint_url="https://api-inference.huggingface.co/models/your-model-name",
42
  huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
43
  model_kwargs={"temperature": 0.5, "max_new_tokens": 512}
44
  )
 
38
  # )
39
  # Define the LLM
40
  llm = HuggingFaceEndpoint(
41
+ endpoint_url="https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct",
42
  huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
43
  model_kwargs={"temperature": 0.5, "max_new_tokens": 512}
44
  )