Update app.py

app.py

@@ -4,11 +4,11 @@ from transformers import GPT2Tokenizer, TrainingArguments, Trainer, GPT2LMHeadMo
 
 tokenizer = GPT2Tokenizer.from_pretrained('mindwrapped/gpt2-lotr-fellowship', bos_token='<|startoftext|>',
                                           eos_token='<|endoftext|>', pad_token='<|pad|>')
-model = GPT2LMHeadModel.from_pretrained('mindwrapped/gpt2-lotr-fellowship').cuda()
+model = GPT2LMHeadModel.from_pretrained('mindwrapped/gpt2-lotr-fellowship')
 
 
 def generate_text(text, temperature):
-  generated = tokenizer("<|startoftext|> " + text, return_tensors="pt").input_ids.cuda()
+  generated = tokenizer("<|startoftext|> " + text, return_tensors="pt").input_ids
   sample_outputs = model.generate(generated, do_sample=True, top_k=50, 
                                   max_length=300, top_p=0.95, temperature=float(temperature), num_return_sequences=1)