Update app.py

app.py

@@ -8,7 +8,7 @@ from typing import List
 import torch
 # Load the model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
-model = AutoModelForCausalLM.from_pretrained("gpt2",device_map='cuda')
+model = AutoModelForCausalLM.from_pretrained("gpt2")
 
 generation_config = model.generation_config
 generation_config.temperature = 0