Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 import transformers
 from transformers import GPT2LMHeadModel, GPT2Tokenizer, Trainer, TrainingArguments, DataCollatorForLanguageModeling
 from datasets import Dataset
-import spaces
+# import spaces
 
 # Set random seeds for reproducibility
 random.seed(42)
@@ -59,7 +59,7 @@ def prepare_dataset(data, tokenizer, block_size=128):
     tokenized_dataset.set_format(type='torch', columns=['input_ids', 'attention_mask', 'labels'])
     return tokenized_dataset
 
-@spaces.GPU()
+# @spaces.GPU()
 def fitness_function(individual, train_dataset, tokenizer, model_state_dict):
     # Initialize the model inside this function
     model = GPT2LMHeadModel.from_pretrained('gpt2')