Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr 
 import pandas as pd
+import spaces
 from transformers import RagTokenizer, RagRetriever, RagTokenForGeneration
 
 # Load the tokenizer and retriever
@@ -14,13 +15,14 @@ inputs = tokenizer(contexts, return_tensors='pt', padding=True, truncation=True)
 labels = tokenizer(responses, return_tensors='pt', padding=True, truncation=True)
 
 # Load your dataset
-df = pd.read_csv('your_dataset.csv')
+df = pd.read_csv('10kstats.csv')
 
 # Ensure the dataset has the required columns for RAG
 # For example, it should have 'context' and 'response' columns
 contexts = df['Abstract'].tolist()
 #responses = df['response'].tolist()
 
+@spaces.GPU
 def generate_response(input_text):
     input_ids = tokenizer([input_text], return_tensors='pt')['input_ids']
     outputs = model.generate(input_ids)