Update app.py

app.py

@@ -12,10 +12,10 @@ model = AutoModelForCausalLM.from_pretrained(model_dir)
 
 code_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-prompt = "def quicksort(arr):"
+prompt = "def "
 
 
-inputs_text=st.text_input("Please enter the text",value='v="prem kumar" for i in v :')
+inputs_text=st.text_input("Please enter the text",value="write a print statement")
 
 if st.button("submit"):
     generated_code = code_generator(inputs_text, max_length=200, num_return_sequences=1)