Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import torch
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from summarizer import Summarizer
 
@@ -20,7 +21,6 @@ if choice == 'NFT':
     if st.button("Generate"):
         prompt = "<|startoftext|>"
         generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
-        generated = generated.to(device)
         sample_outputs = model.generate(
         generated,
         do_sample=True,