Update app.py

app.py

@@ -19,7 +19,7 @@ def get_model():
  
 model, tokenizer = get_model()
 #g = 
-#c = 5
+c = 5
 with st.form(key='my_form'):
     prompt = st.text_input('Enter sentence', '')
     #prompt = st.text_area(label='Enter Number of words', value=c)
@@ -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 = best_words[0]              
+        final_string = prompt + best_words[0:5]              
         st.write(final_string)