Update app.py

app.py

@@ -21,22 +21,14 @@ if choice == 'NFT':
     if st.button("Generate"):
         prompt = "<|startoftext|>"
         generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)        
-        sample_outputs = model.generate(
-        generated,
-        do_sample=True,
-        top_k=50,
-        max_length = 512,
-        top_p=0.95,
-        num_return_sequences=1
-        )
-    for i, sample_output in enumerate(sample_outputs):
-        generated_text = tokenizer.decode(sample_output, skip_special_tokens=True)
+        output = model.generate(generated, do_sample=True, top_k=50, max_length = 512, top_p=0.95, num_return_sequences=1)
+        generated_text = tokenizer.decode(output, skip_special_tokens=True)
         summary = summarize(generated_text, num_sentences=1)
       
-    #st.text("Keywords: {}\n".format(keywords))
-    #st.text("Length in number of words: {}\n".format(length))
-    st.text("This is your tailored blog article {generated_text}")
-    st.text("This is a tweet-sized summary of your article {summary}")
+        #st.text("Keywords: {}\n".format(keywords))
+        #st.text("Length in number of words: {}\n".format(length))
+        st.text("This is your tailored blog article:", generated_text)
+        st.text("This is a tweet-sized summary of your article: ", summary)
 else:
     st.write("Topic not available yet")