Update app.py

app.py

@@ -10,6 +10,7 @@ from model import create_model
 
 api_key = st.secrets["API"]
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model,tokenizer = create_model()
 # Load existing ideas from a file