Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ from transformers import AutoTokenizer
|
|
| 52 |
st.set_page_config(page_title="Document Analysis", page_icon=":robot:")
|
| 53 |
st.header("Chat with your document 📄 (Model: Falcon-7B-Instruct)")
|
| 54 |
|
| 55 |
-
model_name = "tiiuae/falcon-
|
| 56 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 57 |
|
| 58 |
class CustomHuggingFaceEndpoint(HuggingFaceHub):
|
|
|
|
| 52 |
st.set_page_config(page_title="Document Analysis", page_icon=":robot:")
|
| 53 |
st.header("Chat with your document 📄 (Model: Falcon-7B-Instruct)")
|
| 54 |
|
| 55 |
+
model_name = "tiiuae/falcon-40b-instruct"
|
| 56 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 57 |
|
| 58 |
class CustomHuggingFaceEndpoint(HuggingFaceHub):
|