Commit
β’
18e8d0d
1
Parent(s):
8a27771
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-7b-instruct"
|
56 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
57 |
|
58 |
class CustomHuggingFaceEndpoint(HuggingFaceHub):
|