gourisankar85 commited on
Commit
e5db4c0
·
verified ·
1 Parent(s): 3ce6662

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -3,6 +3,11 @@ import PyPDF2
3
  import faiss
4
  import numpy as np
5
 
 
 
 
 
 
6
  #Specify file paths
7
  file_path1 = './AST-1.pdf'
8
  file_path2 = './AST-2.pdf'
 
3
  import faiss
4
  import numpy as np
5
 
6
+ # Load the LLM for generation
7
+ generation_model_name = 'facebook/bart-large-cnn'
8
+ generation_model = AutoModelForSeq2SeqLM.from_pretrained(generation_model_name)
9
+ tokenizer = AutoTokenizer.from_pretrained(generation_model_name)
10
+
11
  #Specify file paths
12
  file_path1 = './AST-1.pdf'
13
  file_path2 = './AST-2.pdf'