Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import numpy as np
 from transformers import pipeline, BertTokenizer, BertModel
 import faiss
+import spaces
 
 # Load CSV data
 data = pd.read_csv('RBDx10stats.csv')
@@ -29,6 +30,7 @@ tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
 model = BertModel.from_pretrained('bert-base-uncased')
 
 # Function to retrieve the relevant document and generate a response
+@spaces.GPU(duration=120)
 def retrieve_and_generate(question):
     # Embed the question
     question_embedding = embed_question(question, model, tokenizer)