Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,10 @@ from huggingface_hub.inference_api import InferenceApi as InferenceClient
|
|
| 5 |
|
| 6 |
import streamlit as st
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
from prompts import (
|
| 9 |
ACTION_PROMPT,
|
| 10 |
ADD_PROMPT,
|
|
|
|
| 5 |
|
| 6 |
import streamlit as st
|
| 7 |
|
| 8 |
+
# Load LLM
|
| 9 |
+
llm = HuggingFaceHub(repo_id="tiiuae/falcon-7b-instruct", model_kwargs={"temperature": 0.1, "max_new_tokens": 500})
|
| 10 |
+
|
| 11 |
+
|
| 12 |
from prompts import (
|
| 13 |
ACTION_PROMPT,
|
| 14 |
ADD_PROMPT,
|