Update app.py

app.py

@@ -6,6 +6,9 @@ import json
 from transformers import AutoTokenizer, AutoModelForQuestionAnswering, AutoModelForSequenceClassification
 import spaces
 
+# Ensure you have GPU support
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+
 # Load the CSV file with embeddings
 df = pd.read_csv('RBDx10kstats.csv')
 df['embedding'] = df['embedding'].apply(json.loads)  # Convert JSON string back to list