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