Spaces:
Build error
Build error
Commit
·
7a37089
1
Parent(s):
a94ac4c
Update app.py
Browse files
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,
|