nvinay1803 commited on
Commit
18e8d0d
β€’
1 Parent(s): 8a27771

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-40b-instruct"
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):