jcarbonnell commited on
Commit
b6583b8
·
1 Parent(s): 5cc1934

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -20,10 +20,7 @@ if choice == 'NFT':
20
  length=st.text_area("How long should be your text? (default: 512 words)")
21
 
22
  if st.button("Generate"):
23
- prompt = "<|startoftext|>"
24
- generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
25
- output = model.generate(generated, do_sample=True, top_k=50, max_length = 512, top_p=0.95, num_return_sequences=1)
26
- generated_text = tokenizer.decode(output, skip_special_tokens=True)
27
  #summary = summarize(generated_text, num_sentences=1)
28
 
29
  #st.text("Keywords: {}\n".format(keywords))
 
20
  length=st.text_area("How long should be your text? (default: 512 words)")
21
 
22
  if st.button("Generate"):
23
+ model("", num_return_sequences=1)
 
 
 
24
  #summary = summarize(generated_text, num_sentences=1)
25
 
26
  #st.text("Keywords: {}\n".format(keywords))