Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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/
|
| 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 |
)
|