Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ class BasicAgent: # Some times Inheritance is needed
|
|
| 32 |
#llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # ChatML understands roles (user, assistant, system)
|
| 33 |
# Define the LLM
|
| 34 |
llm = HuggingFaceHub(
|
| 35 |
-
repo_id="
|
| 36 |
model_kwargs={"temperature": 0.5, "max_new_tokens": 512}
|
| 37 |
)
|
| 38 |
|
|
|
|
| 32 |
#llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # ChatML understands roles (user, assistant, system)
|
| 33 |
# Define the LLM
|
| 34 |
llm = HuggingFaceHub(
|
| 35 |
+
repo_id="tiiuae/falcon-7b-instruct",
|
| 36 |
model_kwargs={"temperature": 0.5, "max_new_tokens": 512}
|
| 37 |
)
|
| 38 |
|