Update app.py

app.py

@@ -26,7 +26,7 @@ def generate_text(input_text, max_length=16, num_beams=5, do_sample=False, no_re
     - generated_text (str): The generated text.
     """
     # Encode the input text and move it to the appropriate device
-    input_ids = tokenizer(input_text, return_tensors='pt')['input_ids'].to(device)
+    input_ids = tokenizer(input_text, return_tensors='pt')['input_ids']
     # Generate text using the model
     output = model.generate(input_ids, max_length=max_length, num_beams=num_beams,
                             do_sample=do_sample, no_repeat_ngram_size=no_repeat_ngram_size)
@@ -49,7 +49,7 @@ def generate_text_with_nucleus_search(input_text, max_length=16, do_sample=True,
     - generated_text (str): The generated text.
     """
     # Encode the input text and move it to the appropriate device
-    input_ids = tokenizer(input_text, return_tensors='pt')['input_ids'].to(device)
+    input_ids = tokenizer(input_text, return_tensors='pt')['input_ids']
     # Generate text using nucleus sampling
     output = model.generate(input_ids, max_length=max_length, do_sample=do_sample, top_p=top_p)
     # Decode the generated output