Update app.py

app.py

@@ -11,12 +11,12 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2', use_auth_token=HF_TOKEN)
 model = GPT2LMHeadModel.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
 model.eval()
-model.to('cuda')
+model.to("cpu")
 
 # Define the function that generates text from a prompt
 def generate_text(prompt):
     input_tokens = tokenizer.encode(prompt, return_tensors='pt')
-    input_tokens = input_tokens.to('cuda')
+    input_tokens = input_tokens.to('cpu')
     
     generated_tokens = []