Update app.py

app.py

@@ -4,8 +4,8 @@ import torch
 
 trained_tokenizer = GPT2Tokenizer.from_pretrained("Kumarkishalaya/GPT-2-next-word-prediction")
 trained_model = GPT2LMHeadModel.from_pretrained("Kumarkishalaya/GPT-2-next-word-prediction")
-untrained_model = GPT2Tokenizer.from_pretrained("gpt2")
-untrained_tokenizer = ("gpt2")
+untrained_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+untrained_model = GPT2LMHeadModel.from_pretrained("gpt2")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 trained_model.to(device)
 untrained_model.to(device)