updating zeroGPU

app.py

@@ -50,18 +50,10 @@ dataset = load_dataset('tosin2013/autogen', streaming=True)
 dataset = Dataset.from_list(list(dataset['train']))
 
 # Check GPU availability and initialize embeddings
-if torch.cuda.is_available():
-    print("[INFO] GPU is available, using CUDA")
-    device = "cuda"
-    # Initialize CUDA device
-    torch.cuda.init()
-else:
-    print("[INFO] GPU not available, using CPU")
-    device = "cpu"
-
+# Use CPU for embeddings since GPU is handled by spaces.GPU decorator
 embeddings = HuggingFaceEmbeddings(
     model_name="sentence-transformers/all-MiniLM-L6-v2",
-    model_kwargs={"device": device}
+    model_kwargs={"device": "cpu"}
 )
 
 # Extract texts from the dataset