2.19 kB
| ## How to use | |
| ```python | |
| import logging | |
| from simpletransformers.seq2seq import Seq2SeqModel, Seq2SeqArgs | |
| logging.basicConfig(level=logging.INFO) | |
| transformers_logger = logging.getLogger("transformers") | |
| transformers_logger.setLevel(logging.WARNING) | |
| model_args = Seq2SeqArgs() | |
| # 加载本地训练好的模型 | |
| model = Seq2SeqModel( | |
| encoder_decoder_type="bart", | |
| encoder_decoder_name="NTUYG/SOTitle-java-BART", | |
| args=model_args, | |
| ) | |
| describe = """ | |
| I am a beginner at Android Java development but I have a few years of school + uni experience in Java. I am trying to write to a text file in an assets folder in my app using FileOutputStream but it doesn't seem to write to it at all since I am using InputStream to read the file after and there haven't any updates. Here is my code | |
| """ | |
| code = """ | |
| private void updateTextFile(String update) { | |
| FileOutputStream fos = null; | |
| try | |
| { | |
| fos = openFileOutput("Questions",MODE_PRIVATE); | |
| fos.write("Testing".getBytes()); | |
| } | |
| catch (FileNotFoundException e) | |
| { | |
| e.printStackTrace(); | |
| } | |
| catch (IOException e) | |
| { | |
| e.printStackTrace(); | |
| } | |
| finally | |
| { | |
| if(fos!=null) | |
| { | |
| try | |
| { | |
| fos.close(); | |
| } | |
| catch (IOException e) | |
| { | |
| e.printStackTrace(); | |
| } | |
| } | |
| } | |
| String text = ""; | |
| try | |
| { | |
| InputStream is = getAssets().open("Questions"); | |
| int size = is.available(); | |
| byte[] buffer = new byte[size]; | |
| is.read(buffer); | |
| is.close(); | |
| text = new String(buffer); | |
| } | |
| catch (IOException e) | |
| { | |
| e.printStackTrace(); | |
| } | |
| System.out.println("Tesing output " + text); | |
| } | |
| """ | |
| from nltk import word_tokenize | |
| describe = describe.replace('\n',' ').replace('\r',' ') | |
| describe = ' '.join(word_tokenize(describe)) | |
| code = code.replace('\n',' ').replace('\r',' ') | |
| code = ' '.join(word_tokenize(code)) | |
| # human ： Java Android Cant seem to update text file using FileOutputStream | |
| body = describe + ' <code> ' + code +' </code>' | |
| print( | |
| model.predict( | |
| [ | |
| body | |
| ] | |
| ) | |
| ) | |
| ``` | |