Update app.py

app.py

@@ -67,7 +67,7 @@ class MemoryEfficientNN(nn.Module):
 class MemoryEfficientDataset(IterableDataset):
     def __init__(self, X, y, batch_size):
         self.X = X
-        self.y = y
+        self.y = y.astype(int)  # Convert labels to integers
         self.batch_size = batch_size
 
     def __iter__(self):
@@ -76,6 +76,12 @@ class MemoryEfficientDataset(IterableDataset):
             y_batch = self.y[i:i+self.batch_size]
             yield torch.FloatTensor(X_batch), torch.LongTensor(y_batch)
 
+    def __iter__(self):
+        for i in range(0, len(self.y), self.batch_size):
+            X_batch = self.X[i:i+self.batch_size].toarray()
+            y_batch = self.y[i:i+self.batch_size]
+            yield torch.FloatTensor(X_batch), torch.LongTensor(y_batch)
+
 # Train Memory-Efficient Neural Network
 X_train, X_test, y_train, y_test = train_test_split(contexts_encoded, emotions_target, test_size=0.2, random_state=42)
 input_size = X_train.shape[1]