Update app.py

app.py

@@ -10,12 +10,18 @@ translator = pipeline('translation', model=model_checkpoint, decoder_start_token
 st.title('NLTM')
 
 st.header('Hindi-Sanskrit Translation Model')
+submit = st.button('Translate')  
 
 query = st.text_input("Enter the Input Sentence", "")
-
 input_sentence = query + ' </s>' + ' <2hi>'
-output_sentence = translator(input_sentence,max_length=128)
-output_sentence = output_sentence[0]["translation_text"]
-output_sentence = output_sentence.replace('<2sa>','')
 
-st.write('Sanskrit Sentence:',output_sentence)
+if submit:
+  st.subheader('Sanskrit Sentence')
+  
+  with st.spinner(text='This may take a moment...'):
+    output_sentence = translator(input_sentence,max_length=128)
+    
+  output_sentence = output_sentence[0]["translation_text"]
+  output_sentence = output_sentence.replace('<2sa>','')
+    
+  st.write(output_sentence)