jcarbonnell commited on
Commit
7a37089
·
1 Parent(s): a94ac4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import streamlit as st
 
2
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
3
  from summarizer import Summarizer
4
 
@@ -20,7 +21,6 @@ if choice == 'NFT':
20
  if st.button("Generate"):
21
  prompt = "<|startoftext|>"
22
  generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
23
- generated = generated.to(device)
24
  sample_outputs = model.generate(
25
  generated,
26
  do_sample=True,
 
1
  import streamlit as st
2
+ import torch
3
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
4
  from summarizer import Summarizer
5
 
 
21
  if st.button("Generate"):
22
  prompt = "<|startoftext|>"
23
  generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
 
24
  sample_outputs = model.generate(
25
  generated,
26
  do_sample=True,