abakerdp commited on
Commit
91770df
·
verified ·
1 Parent(s): 76da4a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import pinecone
3
  from sentence_transformers import SentenceTransformer
4
- from transformers import AutoTokenizer, AutoModelForSeq2SeqGeneration
5
  import torch
6
  import PyPDF2
7
  import io
@@ -11,7 +11,7 @@ from tqdm import tqdm
11
  # Initialize models and Pinecone
12
  embeddings_model = SentenceTransformer('all-MiniLM-L6-v2')
13
  tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
14
- model = AutoModelForSeq2SeqGeneration.from_pretrained("google/flan-t5-base")
15
 
16
  # Initialize Pinecone with environment variable
17
  PINECONE_API_KEY = os.getenv('PINECONE_API_KEY')
 
1
  import gradio as gr
2
  import pinecone
3
  from sentence_transformers import SentenceTransformer
4
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM # Fixed this line
5
  import torch
6
  import PyPDF2
7
  import io
 
11
  # Initialize models and Pinecone
12
  embeddings_model = SentenceTransformer('all-MiniLM-L6-v2')
13
  tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
14
+ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base") # And this line
15
 
16
  # Initialize Pinecone with environment variable
17
  PINECONE_API_KEY = os.getenv('PINECONE_API_KEY')