Update app.py

app.py

@@ -22,7 +22,7 @@ g = "life is a"
 c = 5
 with st.form(key='my_form'):
     prompt = st.text_area(label='Enter sentence', value=g)
-    prompt = st.text_area(label='Enter Number of words', value=c)
+    #prompt = st.text_area(label='Enter Number of words', value=c)
     submit_button = st.form_submit_button(label='Submit')
     if submit_button:
       with torch.no_grad():
@@ -38,5 +38,5 @@ with st.form(key='my_form'):
         text.append(best_indices[0].item())
         best_probabilities = probabilities[best_indices].tolist()
         words = []
-        final_string = ' '.join(best_words[c])              
+        final_string = ' '.join(best_words)              
         st.write(final_string)