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

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@
2
  import PyPDF2
3
  import faiss
4
  import numpy as np
 
5
 
6
  # Load the LLM for generation
7
  generation_model_name = 'facebook/bart-large-cnn'
 
2
  import PyPDF2
3
  import faiss
4
  import numpy as np
5
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
6
 
7
  # Load the LLM for generation
8
  generation_model_name = 'facebook/bart-large-cnn'