Update app.py

app.py

@@ -22,6 +22,6 @@ st.info("Max generated sentence: 100 words")
 if (st.button("Generate")):
     input_ids = tokenizer.encode(sentence, return_tensors='pt').to(device)
     paragraph_generated = model.generate(input_ids, max_length=100, num_beams=5, no_repeat_ngram_size=2, early_stopping=True).to(device)
-    text = tokenizer.decode(paragraph_generated[0], skip_special_tokens=True).to(device)
+    text = tokenizer.decode(paragraph_generated[0], skip_special_tokens=True)
 
     st.write(text)