Update app.py

app.py

@@ -30,6 +30,9 @@ from transformers import T5ForConditionalGeneration,T5Tokenizer
 summary_model = T5ForConditionalGeneration.from_pretrained('t5-base')
 summary_tokenizer = T5Tokenizer.from_pretrained('t5-base')
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+summary_model = summary_model.to(device)
+
 question_model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_squad_v1')
 question_tokenizer = T5Tokenizer.from_pretrained('ramsrigouthamg/t5_squad_v1')
 question_model = question_model.to(device)