Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 st.title( 'Text generation by GPT model' )
 st.subheader( 'This application shows the difference in text generation between a rugpt3small model trained on general documents and the same model trained on jokes' )
 
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
+device = 'cpu' if torch.cuda.is_available() else 'cpu'
 
 # Load the model tokenizer
 from transformers import GPT2Tokenizer